Merge pull request #608 from FlorentinD/fix-non-enterprise-tests

Fix tests failing against community GDS

Author: FlorentinD

graphdatascience/tests/integration/test_graph_construct.py

@@ -521,11 +521,15 @@ def test_graph_alpha_construct_backward_compat_without_arrow(gds_without_arrow:
 
 
 @pytest.mark.skip_on_aura  # No warning produced when running on Aura
-def test_nodes_only_without__arrow(gds_without_arrow: GraphDataScience) -> None:
+def test_nodes_only_without_arrow(gds_without_arrow: GraphDataScience) -> None:
     nodes = DataFrame({"nodeId": [0], "labels": ["person"]})
     relationships = DataFrame({"sourceNodeId": [], "targetNodeId": [], "relationshipType": []})
 
-    with pytest.warns(UserWarning):
+    if gds_without_arrow.is_licensed():
+        # on enterprise users should use arrow for construct
+        with pytest.warns(UserWarning):
+            G = gds_without_arrow.graph.construct(graph_name="my_graph", nodes=nodes, relationships=relationships)
+    else:
         G = gds_without_arrow.graph.construct(graph_name="my_graph", nodes=nodes, relationships=relationships)
 
     try:

graphdatascience/tests/integration/test_graph_ops.py

@@ -418,6 +418,7 @@ def test_graph_nodeProperty_stream_raise_error_with_duplicate_keys(gds: GraphDat
         gds.graph.nodeProperty.stream(G, "x", db_node_properties=["x", "z", "name"], concurrency=2)
 
 
+@pytest.mark.enterprise
 @pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 2, 0))
 def test_graph_nodeProperties_stream_with_arrow(gds: GraphDataScience) -> None:
     G, _ = gds.graph.project(GRAPH_NAME, {"Node": {"properties": ["x", "y"]}}, "*")
@@ -441,6 +442,7 @@ def test_graph_nodeProperties_stream_with_arrow(gds: GraphDataScience) -> None:
     assert {e for e in name_values["propertyValue"]} == {"nodeA", "nodeB", "nodeC"}
 
 
+@pytest.mark.enterprise
 @pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 5, 0))
 def test_graph_nodeProperties_stream_listNodeLabels_with_arrow(gds: GraphDataScience) -> None:
     G, _ = gds.graph.project(GRAPH_NAME, {"Node": {"properties": ["x"]}}, "*")

graphdatascience/tests/integration/test_system_ops.py

@@ -109,4 +109,6 @@ def test_restore(gds: GraphDataScience) -> None:
 @pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 5, 0))
 def test_deprecated_endpoints(gds: GraphDataScience) -> None:
     gds.beta.listProgress()
-    gds.alpha.systemMonitor()
+
+    if gds.is_licensed():
+        gds.alpha.systemMonitor()