Mark arrow write tests as db_integration

Author: DarthMax

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:

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:

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:

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")

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")

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:

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")

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:

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)

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,