Skip to content

Commit b6e71f9

Browse files
committed
Shorten long lines
1 parent 1dbdaa3 commit b6e71f9

File tree

2 files changed

+9
-12
lines changed

2 files changed

+9
-12
lines changed

graphdatascience/tests/unit/test_aura_db_query_runner.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,7 @@ def test_extracts_parameters_algo_write() -> None:
7474
"config": {"jobId": "my-job", "writeToResultStore": True},
7575
}
7676
assert (
77-
db_query_runner.last_query()
78-
== "CALL gds.arrow.write($graphName, $databaseName, $jobId, $arrowConfiguration)"
77+
db_query_runner.last_query() == "CALL gds.arrow.write($graphName, $databaseName, $jobId, $arrowConfiguration)"
7978
)
8079
assert db_query_runner.last_params() == {
8180
"graphName": "g",
@@ -106,8 +105,7 @@ def test_arrow_and_write_configuration() -> None:
106105
"config": {"writeToResultStore": True, "jobId": "my-job"},
107106
}
108107
assert (
109-
db_query_runner.last_query()
110-
== "CALL gds.arrow.write($graphName, $databaseName, $jobId, $arrowConfiguration)"
108+
db_query_runner.last_query() == "CALL gds.arrow.write($graphName, $databaseName, $jobId, $arrowConfiguration)"
111109
)
112110
assert db_query_runner.last_params() == {
113111
"graphName": "g",
@@ -151,8 +149,7 @@ def test_arrow_and_write_configuration_graph_write() -> None:
151149
"config": {"writeToResultStore": True, "jobId": "my-job"},
152150
}
153151
assert (
154-
db_query_runner.last_query()
155-
== "CALL gds.arrow.write($graphName, $databaseName, $jobId, $arrowConfiguration)"
152+
db_query_runner.last_query() == "CALL gds.arrow.write($graphName, $databaseName, $jobId, $arrowConfiguration)"
156153
)
157154
assert db_query_runner.last_params() == {
158155
"graphName": "g",

graphdatascience/tests/unit/test_graph_ops.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,9 @@ def test_project_remote(runner: CollectingQueryRunner, aura_gds: AuraGraphDataSc
9595
aura_gds.graph.project("g", "RETURN gds.graph.project.remote(0, 1, null)")
9696

9797
assert (
98-
runner.last_query()
99-
== "CALL gds.arrow.project("
100-
+ "$graph_name, $query, $concurrency, $undirected_relationship_types, $inverse_indexed_relationship_types, $arrow_configuration)"
98+
runner.last_query() == "CALL gds.arrow.project("
99+
"$graph_name, $query, $concurrency, "
100+
"$undirected_relationship_types, $inverse_indexed_relationship_types, $arrow_configuration)"
101101
)
102102
# injection of token and host into the params is done by the actual query runner
103103
assert runner.last_params() == {
@@ -721,9 +721,9 @@ def test_remote_projection_all_configuration(runner: CollectingQueryRunner, aura
721721
)
722722

723723
assert (
724-
runner.last_query()
725-
== "CALL gds.arrow.project("
726-
+ "$graph_name, $query, $concurrency, $undirected_relationship_types, $inverse_indexed_relationship_types, $arrow_configuration)"
724+
runner.last_query() == "CALL gds.arrow.project("
725+
"$graph_name, $query, $concurrency, "
726+
"$undirected_relationship_types, $inverse_indexed_relationship_types, $arrow_configuration)"
727727
)
728728

729729
assert runner.last_params() == {

0 commit comments

Comments
 (0)