Skip to content

Commit a081a70

Browse files
committed
Fix tests
1 parent d313f46 commit a081a70

File tree

4 files changed

+11
-30
lines changed

4 files changed

+11
-30
lines changed

graphdatascience/procedure_surface/cypher/modularity_optimization_cypher_endpoints.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def mutate(
6262

6363
params = CallParameters(
6464
graph_name=G.name(),
65-
configuration=config,
65+
config=config,
6666
)
6767
params.ensure_job_id_in_config()
6868

@@ -108,7 +108,7 @@ def stats(
108108

109109
params = CallParameters(
110110
graph_name=G.name(),
111-
configuration=config,
111+
config=config,
112112
)
113113
params.ensure_job_id_in_config()
114114

@@ -154,7 +154,7 @@ def stream(
154154

155155
params = CallParameters(
156156
graph_name=G.name(),
157-
configuration=config,
157+
config=config,
158158
)
159159
params.ensure_job_id_in_config()
160160

@@ -206,7 +206,7 @@ def write(
206206

207207
params = CallParameters(
208208
graph_name=G.name(),
209-
configuration=config,
209+
config=config,
210210
)
211211
params.ensure_job_id_in_config()
212212

graphdatascience/procedure_surface/cypher/sllpa_cypher_endpoints.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def mutate(
5656

5757
params = CallParameters(
5858
graph_name=G.name(),
59-
configuration=config,
59+
config=config,
6060
)
6161
params.ensure_job_id_in_config()
6262

@@ -94,7 +94,7 @@ def stats(
9494

9595
params = CallParameters(
9696
graph_name=G.name(),
97-
configuration=config,
97+
config=config,
9898
)
9999
params.ensure_job_id_in_config()
100100

@@ -132,7 +132,7 @@ def stream(
132132

133133
params = CallParameters(
134134
graph_name=G.name(),
135-
configuration=config,
135+
config=config,
136136
)
137137
params.ensure_job_id_in_config()
138138

@@ -174,7 +174,7 @@ def write(
174174

175175
params = CallParameters(
176176
graph_name=G.name(),
177-
configuration=config,
177+
config=config,
178178
)
179179
params.ensure_job_id_in_config()
180180

graphdatascience/session/session_v2_endpoints.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,13 @@
22

33
from graphdatascience.arrow_client.authenticated_flight_client import AuthenticatedArrowClient
44
from graphdatascience.arrow_client.v2.remote_write_back_client import RemoteWriteBackClient
5-
from graphdatascience.procedure_surface.api.community.local_clustering_coefficient_endpoints import (
6-
LocalClusteringCoefficientEndpoints,
7-
)
85
from graphdatascience.procedure_surface.api.community.clique_counting_endpoints import CliqueCountingEndpoints
96
from graphdatascience.procedure_surface.api.community.kmeans_endpoints import KMeansEndpoints
107
from graphdatascience.procedure_surface.api.community.labelpropagation_endpoints import LabelPropagationEndpoints
118
from graphdatascience.procedure_surface.api.community.leiden_endpoints import LeidenEndpoints
9+
from graphdatascience.procedure_surface.api.community.local_clustering_coefficient_endpoints import (
10+
LocalClusteringCoefficientEndpoints,
11+
)
1212
from graphdatascience.procedure_surface.api.community.maxkcut_endpoints import MaxKCutEndpoints
1313
from graphdatascience.procedure_surface.api.community.modularity_optimization_endpoints import (
1414
ModularityOptimizationEndpoints,

graphdatascience/tests/integrationV2/procedure_surface/cypher/test_sllpa_cypher_endpoints.py

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -102,25 +102,6 @@ def test_sllpa_write(sllpa_endpoints: SllpaCypherEndpoints, sample_graph: GraphV
102102
assert isinstance(result.configuration, dict)
103103

104104

105-
def test_sllpa_mutate_with_parameters(sllpa_endpoints: SllpaCypherEndpoints, sample_graph: GraphV2) -> None:
106-
"""Test SLLPA mutate operation with various parameters."""
107-
result = sllpa_endpoints.mutate(
108-
G=sample_graph,
109-
mutate_property="sllpa_community_advanced",
110-
min_association_strength=0.1,
111-
concurrency=1,
112-
max_iterations=1,
113-
)
114-
115-
assert result.ran_iterations > 0
116-
assert result.did_converge in [True, False]
117-
assert result.compute_millis > 0
118-
assert result.pre_processing_millis >= 0
119-
assert result.mutate_millis >= 0
120-
assert result.node_properties_written == 6
121-
assert isinstance(result.configuration, dict)
122-
123-
124105
def test_sllpa_estimate(sllpa_endpoints: SllpaCypherEndpoints, sample_graph: GraphV2) -> None:
125106
"""Test SLLPA estimate operation."""
126107
result = sllpa_endpoints.estimate(sample_graph, max_iterations=1)

0 commit comments

Comments
 (0)