Added Support for numpy ndarrays and scalars

Author: BorisTasevski

gqlalchemy/utilities.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 
 import math
+import numpy as np
+
 from datetime import datetime, date, time, timedelta
 from enum import Enum
 from typing import Any, Dict, List, Optional, Tuple, Union
@@ -88,13 +90,16 @@ def to_cypher_value(value: Any, config: NetworkXCypherConfig = None) -> str:
     if value_type in [int, float, bool]:
         return str(value)
 
-    if value_type in [list, set, tuple]:
+    if value_type in [list, set, tuple, np.ndarray]:
         return f"[{', '.join(map(to_cypher_value, value))}]"
 
     if value_type == dict:
         lines = ", ".join(f"{k}: {to_cypher_value(v)}" for k, v in value.items())
         return f"{{{lines}}}"
 
+    if isinstance(value, np.generic):
+        return to_cypher_value(value.item())
+
     if value is None:
         return "null"
 

tests/test_utilities.py

@@ -14,6 +14,7 @@
 
 import datetime
 import math
+import numpy as np
 import pytest
 
 from gqlalchemy.utilities import (
@@ -37,6 +38,10 @@
         ({"k1": 123, "k2": {"subkey1": "abc"}}, "{k1: 123, k2: {subkey1: 'abc'}}"),
         (None, "null"),
         (True, "True"),
+        (np.array([1, 2, 3]), "[1, 2, 3]"),
+        (np.array([1.23, 2.34, 3.45]), "[1.23, 2.34, 3.45]"),
+        (np.array(["1", "2", "3"]), "['1', '2', '3']"),
+        (np.array([[1, 2], [3, 4], [4, 5]]), "[[1, 2], [3, 4], [4, 5]]"),
     ],
 )
 def test_to_cypher_value(value, cypher_value):