Fix GDS intregation test for arrow

Co-authored-by: Adam Schill Collberg <adam.schill.collberg@protonmail.com>

Author: FlorentinD

python-wrapper/tests/conftest.py

@@ -39,7 +39,6 @@ def gds() -> Generator[Any, None, None]:
             auth=(dbms_connection_info.username, dbms_connection_info.password),
             aura_ds=True,
             database="neo4j",
-            arrow=False,
         )
 
         api.delete_instance(id)

python-wrapper/tests/gds_helper.py

@@ -11,7 +11,7 @@ def connect_to_plugin_gds(uri: str) -> GraphDataScience:
     if os.environ.get("NEO4J_USER"):
         NEO4J_AUTH = (os.environ.get("NEO4J_USER", "DUMMY"), os.environ.get("NEO4J_PASSWORD", "neo4j"))
 
-    return GraphDataScience(endpoint=uri, auth=NEO4J_AUTH, database="neo4j", arrow=False)
+    return GraphDataScience(endpoint=uri, auth=NEO4J_AUTH, database="neo4j")
 
 
 def aura_api() -> AuraApi:

python-wrapper/tests/test_gds.py

@@ -16,7 +16,7 @@ def test_from_gds_integration(gds: Any) -> None:
     nodes = pd.DataFrame(
         {
             "nodeId": [0, 1, 2],
-            "labels": ["A", "C", ["A", "B"]],
+            "labels": [["A"], ["C"], ["A", "B"]],
             "score": [1337, 42, 3.14],
             "component": [1, 4, 2],
         }