Remove usage of removed test class

Author: Mats-SX

graphdatascience/tests/integrationV2/procedure_surface/session/test_walking_skeleton.py

@@ -6,9 +6,6 @@
 from graphdatascience.arrow_client.authenticated_flight_client import AuthenticatedArrowClient
 from graphdatascience.session.aura_graph_data_science import AuraGraphDataScience
 from graphdatascience.session.session_v2_endpoints import SessionV2Endpoints
-from graphdatascience.tests.integrationV2.procedure_surface.arrow.graph_creation_helper import (
-    ArrowGraphForTests,
-)
 
 
 @pytest.fixture(scope="package")
@@ -36,8 +33,11 @@ def setup_db(db_query_runner: QueryRunner) -> Generator[None, None, None]:
 
 @pytest.mark.db_integration
 def test_walking_skeleton(gds: AuraGraphDataScience) -> None:
-    project_result = gds.v2.graph.project("g", "MATCH (n)-->(m) RETURN gds.graph.project.remote(n, m)")
-    G = ArrowGraphForTests(project_result.graph_name)
+    g_and_result = gds.v2.graph.project("g", "MATCH (n)-->(m) RETURN gds.graph.project.remote(n, m)")
+
+    G = g_and_result.graph
+    project_result = g_and_result.result
+    assert project_result.node_count == 2
 
     wcc_mutate_result = gds.v2.wcc.mutate(G, mutate_property="wcc")