Skip to content

Commit 0ec9b6e

Browse files
committed
Python regression.
1 parent 0a7f90c commit 0ec9b6e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pyrasterframes/src/main/python/tests/VectorTypesTests.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,10 +165,11 @@ def test_xz2_index(self):
165165
self.assertSetEqual(indexes, expected)
166166

167167
# Test against proj_raster (has CRS and Extent embedded).
168-
result_one_arg = self.df.select(rf_xz2_index('tile').alias('ix')) \
168+
df = self.spark.read.raster(self.img_uri)
169+
result_one_arg = df.select(rf_xz2_index('proj_raster').alias('ix')) \
169170
.agg(F_min('ix')).first()[0]
170171

171-
result_two_arg = self.df.select(rf_xz2_index(rf_extent('tile'), rf_crs('tile')).alias('ix')) \
172+
result_two_arg = df.select(rf_xz2_index(rf_extent('proj_raster'), rf_crs('proj_raster')).alias('ix')) \
172173
.agg(F_min('ix')).first()[0]
173174

174175
self.assertEqual(result_two_arg, result_one_arg)

0 commit comments

Comments
 (0)