@@ -165,24 +165,20 @@ def test_spatial_join_query_flat_to_df(self):
165
165
right_geometries = self .__row_to_list (right_geometries_raw )
166
166
167
167
# Ignore the ordering of these
168
- assert set (geom [0 ] for geom in left_geometries ) == set (
169
- [
170
- "POLYGON ((0 4, -3 3, -8 6, -6 8, -2 9, 0 4))" ,
171
- "POLYGON ((10 3, 10 6, 14 6, 14 3, 10 3))" ,
172
- "POLYGON ((2 2, 2 4, 3 5, 7 5, 9 3, 8 1, 4 1, 2 2))" ,
173
- "POLYGON ((-1 -1, -1 -3, -2 -5, -6 -8, -5 -2, -3 -2, -1 -1))" ,
174
- "POLYGON ((-1 -1, -1 -3, -2 -5, -6 -8, -5 -2, -3 -2, -1 -1))" ,
175
- ]
176
- )
177
- assert set (geom [0 ] for geom in right_geometries ) == set (
178
- [
179
- "POINT (-3 5)" ,
180
- "POINT (11 5)" ,
181
- "POINT (4 3)" ,
182
- "POINT (-1 -1)" ,
183
- "POINT (-4 -5)" ,
184
- ]
185
- )
168
+ assert {geom [0 ] for geom in left_geometries } == {
169
+ "POLYGON ((0 4, -3 3, -8 6, -6 8, -2 9, 0 4))" ,
170
+ "POLYGON ((10 3, 10 6, 14 6, 14 3, 10 3))" ,
171
+ "POLYGON ((2 2, 2 4, 3 5, 7 5, 9 3, 8 1, 4 1, 2 2))" ,
172
+ "POLYGON ((-1 -1, -1 -3, -2 -5, -6 -8, -5 -2, -3 -2, -1 -1))" ,
173
+ "POLYGON ((-1 -1, -1 -3, -2 -5, -6 -8, -5 -2, -3 -2, -1 -1))" ,
174
+ }
175
+ assert {geom [0 ] for geom in right_geometries } == {
176
+ "POINT (-3 5)" ,
177
+ "POINT (11 5)" ,
178
+ "POINT (4 3)" ,
179
+ "POINT (-1 -1)" ,
180
+ "POINT (-4 -5)" ,
181
+ }
186
182
187
183
def test_range_query_flat_to_df (self ):
188
184
poi_point_rdd = WktReader .readToGeometryRDD (
0 commit comments