Skip to content

Commit fc5d165

Browse files
Mats-SXFlorentinD
andcommitted
Test lastfm with Arrow before testing with Cypher
We had flaky test failures due to some timeout that maybe this helps diagnose. Co-authored-by: Florentin Dörre <florentin.dorre@neo4j.com>
1 parent f7083cc commit fc5d165

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

graphdatascience/tests/integration/test_graph_construct.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -145,10 +145,9 @@ def test_imdb_graph_with_arrow(gds: GraphDataScience) -> None:
145145
G.drop()
146146

147147

148-
@pytest.mark.filterwarnings("ignore: GDS Enterprise users can use Apache Arrow")
149148
@pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 3, 0))
150-
def test_lastfm_graph_without_arrow(gds_without_arrow: GraphDataScience) -> None:
151-
G = gds_without_arrow.graph.load_lastfm()
149+
def test_lastfm_graph_with_arrow(gds: GraphDataScience) -> None:
150+
G = gds.graph.load_lastfm()
152151

153152
try:
154153
assert G.node_count() == 19914
@@ -162,14 +161,14 @@ def test_lastfm_graph_without_arrow(gds_without_arrow: GraphDataScience) -> None
162161
assert set(G.relationship_properties("TAGGED")) == {"day", "month", "year", "tagID", "timestamp"}
163162
assert set(G.relationship_properties("LISTEN_TO")) == {"weight"}
164163
assert set(G.relationship_properties("IS_FRIEND")) == set()
165-
166164
finally:
167165
G.drop()
168166

169167

168+
@pytest.mark.filterwarnings("ignore: GDS Enterprise users can use Apache Arrow")
170169
@pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 3, 0))
171-
def test_lastfm_graph_with_arrow(gds: GraphDataScience) -> None:
172-
G = gds.graph.load_lastfm()
170+
def test_lastfm_graph_without_arrow(gds_without_arrow: GraphDataScience) -> None:
171+
G = gds_without_arrow.graph.load_lastfm()
173172

174173
try:
175174
assert G.node_count() == 19914
@@ -183,6 +182,7 @@ def test_lastfm_graph_with_arrow(gds: GraphDataScience) -> None:
183182
assert set(G.relationship_properties("TAGGED")) == {"day", "month", "year", "tagID", "timestamp"}
184183
assert set(G.relationship_properties("LISTEN_TO")) == {"weight"}
185184
assert set(G.relationship_properties("IS_FRIEND")) == set()
185+
186186
finally:
187187
G.drop()
188188

0 commit comments

Comments
 (0)