Mark arrow write tests as db_integration

DarthMax · DarthMax · commit 6c0e4b4d2ef9 · 2025-09-18T20:22:26.000+02:00
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_articlerank_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_articlerank_arrow_endpoints.py
@@ -90,6 +90,7 @@ def test_articlerank_mutate(articlerank_endpoints: ArticleRankArrowEndpoints, sa
     assert result.node_properties_written == 3
 
 
+@pytest.mark.db_integration
 def test_articlerank_write(
     arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2
 ) -> None:
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_articulationpoints_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_articulationpoints_arrow_endpoints.py
@@ -92,6 +92,7 @@ def test_articulationpoints_stream_not_implemented(
         articulationpoints_endpoints.stream(sample_graph)
 
 
+@pytest.mark.db_integration
 def test_articulationpoints_write(
     arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2
 ) -> None:
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_betweenness_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_betweenness_arrow_endpoints.py
@@ -90,6 +90,7 @@ def test_betweenness_mutate(betweenness_endpoints: BetweennessArrowEndpoints, sa
     assert result.node_properties_written == 3
 
 
+@pytest.mark.db_integration
 def test_betweenness_write(
     arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2
 ) -> None:
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_celf_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_celf_arrow_endpoints.py
@@ -83,6 +83,7 @@ def test_celf_mutate(celf_endpoints: CelfArrowEndpoints, sample_graph: GraphV2)
     assert isinstance(result.configuration, dict)
 
 
+@pytest.mark.db_integration
 def test_celf_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     endpoints = CelfArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
     result = endpoints.write(G=db_graph, seed_set_size=2, write_property="celf_spread")
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_closeness_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_closeness_arrow_endpoints.py
@@ -84,6 +84,7 @@ def test_closeness_mutate(closeness_endpoints: ClosenessArrowEndpoints, sample_g
     assert "p50" in result.centrality_distribution
 
 
+@pytest.mark.db_integration
 def test_closeness_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     endpoints = ClosenessArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
     result = endpoints.write(G=db_graph, write_property="closeness")
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_closeness_harmonic_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_closeness_harmonic_arrow_endpoints.py
@@ -93,6 +93,7 @@ def test_closeness_harmonic_mutate(
     assert "p50" in result.centrality_distribution
 
 
+@pytest.mark.db_integration
 def test_closeness_harmonic_write(
     arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2
 ) -> None:
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_degree_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_degree_arrow_endpoints.py
@@ -81,6 +81,7 @@ def test_degree_mutate(degree_endpoints: DegreeArrowEndpoints, sample_graph: Gra
     assert isinstance(result.configuration, dict)
 
 
+@pytest.mark.db_integration
 def test_degree_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     endpoints = DegreeArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
     result = endpoints.write(G=db_graph, write_property="degree")
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_eigenvector_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_eigenvector_arrow_endpoints.py
@@ -93,6 +93,7 @@ def test_eigenvector_mutate(eigenvector_endpoints: EigenvectorArrowEndpoints, sa
     assert "p50" in result.centrality_distribution
 
 
+@pytest.mark.db_integration
 def test_eigenvector_write(
     arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2
 ) -> None:
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_fastrp_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_fastrp_arrow_endpoints.py
@@ -93,6 +93,7 @@ def test_fastrp_mutate(fastrp_endpoints: FastRPArrowEndpoints, sample_graph: Gra
     assert result.configuration is not None
 
 
+@pytest.mark.db_integration
 def test_fastrp_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     endpoints = FastRPArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
     result = endpoints.write(G=db_graph, write_property="fastrp_embedding", embedding_dimension=32)
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_graphsage_predict_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_graphsage_predict_arrow_endpoints.py
@@ -83,6 +83,7 @@ def test_mutate(gs_model: GraphSageModelV2, sample_graph: GraphV2) -> None:
     assert result.node_properties_written == 4
 
 
+@pytest.mark.db_integration
 def test_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     model, _ = GraphSageTrainArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner)).train(
         G=db_graph,
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_hashgnn_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_hashgnn_arrow_endpoints.py
@@ -99,6 +99,7 @@ def test_hashgnn_stream(hashgnn_endpoints: HashGNNArrowEndpoints, sample_graph:
     assert set(result.columns) == {"nodeId", "embedding"}
 
 
+@pytest.mark.db_integration
 def test_hashgnn_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     """Test HashGNN write operation."""
     endpoints = HashGNNArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_k1coloring_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_k1coloring_arrow_endpoints.py
@@ -92,6 +92,7 @@ def test_k1coloring_mutate(k1coloring_endpoints: K1ColoringArrowEndpoints, sampl
     assert isinstance(result.did_converge, bool)
 
 
+@pytest.mark.db_integration
 def test_k1coloring_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     endpoints = K1ColoringArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
     result = endpoints.write(G=db_graph, write_property="color")
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_kcore_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_kcore_arrow_endpoints.py
@@ -131,6 +131,7 @@ def test_kcore_mutate_with_parameters(kcore_endpoints: KCoreArrowEndpoints, samp
     assert result.node_properties_written == 6
 
 
+@pytest.mark.db_integration
 def test_kcore_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     endpoints = KCoreArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
     result = endpoints.write(G=db_graph, write_property="coreValue")
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_louvain_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_louvain_arrow_endpoints.py
@@ -150,6 +150,7 @@ def test_louvain_stream_with_parameters(louvain_endpoints: LouvainArrowEndpoints
     assert len(result_df.columns) == 2
 
 
+@pytest.mark.db_integration
 def test_louvain_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     """Test Louvain write operation."""
     endpoints = LouvainArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_node2vec_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_node2vec_arrow_endpoints.py
@@ -82,6 +82,7 @@ def test_node2vec_stream(node2vec_endpoints: Node2VecArrowEndpoints, sample_grap
     assert set(result.columns) == {"nodeId", "embedding"}
 
 
+@pytest.mark.db_integration
 def test_node2vec_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     """Test Node2Vec write operation."""
     endpoints = Node2VecArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_pagerank_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_pagerank_arrow_endpoints.py
@@ -91,6 +91,7 @@ def test_pagerank_mutate(pagerank_endpoints: PageRankArrowEndpoints, sample_grap
     assert result.node_properties_written == 3
 
 
+@pytest.mark.db_integration
 def test_pagerank_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     """Test PageRank write operation."""
     endpoints = PageRankArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_scc_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_scc_arrow_endpoints.py
@@ -96,6 +96,7 @@ def test_scc_mutate(scc_endpoints: SccArrowEndpoints, sample_graph: GraphV2) ->
     assert "p10" in result.component_distribution
 
 
+@pytest.mark.db_integration
 def test_scc_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     """Test SCC write operation."""
     endpoints = SccArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
diff --git a/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_wcc_arrow_endpoints.py b/graphdatascience/tests/integrationV2/procedure_surface/arrow/test_wcc_arrow_endpoints.py
@@ -85,6 +85,7 @@ def test_wcc_mutate(wcc_endpoints: WccArrowEndpoints, sample_graph: GraphV2) ->
     assert result.node_properties_written == 3
 
 
+@pytest.mark.db_integration
 def test_wcc_write(arrow_client: AuthenticatedArrowClient, query_runner: QueryRunner, db_graph: GraphV2) -> None:
     endpoints = WccArrowEndpoints(arrow_client, RemoteWriteBackClient(arrow_client, query_runner))
     result = endpoints.write(G=db_graph, write_property="componentId")
