@@ -145,10 +145,9 @@ def test_imdb_graph_with_arrow(gds: GraphDataScience) -> None:
145
145
G .drop ()
146
146
147
147
148
- @pytest .mark .filterwarnings ("ignore: GDS Enterprise users can use Apache Arrow" )
149
148
@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 ()
152
151
153
152
try :
154
153
assert G .node_count () == 19914
@@ -162,14 +161,14 @@ def test_lastfm_graph_without_arrow(gds_without_arrow: GraphDataScience) -> None
162
161
assert set (G .relationship_properties ("TAGGED" )) == {"day" , "month" , "year" , "tagID" , "timestamp" }
163
162
assert set (G .relationship_properties ("LISTEN_TO" )) == {"weight" }
164
163
assert set (G .relationship_properties ("IS_FRIEND" )) == set ()
165
-
166
164
finally :
167
165
G .drop ()
168
166
169
167
168
+ @pytest .mark .filterwarnings ("ignore: GDS Enterprise users can use Apache Arrow" )
170
169
@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 ()
173
172
174
173
try :
175
174
assert G .node_count () == 19914
@@ -183,6 +182,7 @@ def test_lastfm_graph_with_arrow(gds: GraphDataScience) -> None:
183
182
assert set (G .relationship_properties ("TAGGED" )) == {"day" , "month" , "year" , "tagID" , "timestamp" }
184
183
assert set (G .relationship_properties ("LISTEN_TO" )) == {"weight" }
185
184
assert set (G .relationship_properties ("IS_FRIEND" )) == set ()
185
+
186
186
finally :
187
187
G .drop ()
188
188
0 commit comments