Skip to content

Commit 087d349

Browse files
committed
Fix refactoring in raster source test
Signed-off-by: Jason T. Brown <[email protected]>
1 parent 4ccd36d commit 087d349

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828

2929
class RasterSourceTest(TestEnvironment):
3030

31+
3132
@staticmethod
3233
def path(scene, band):
3334
scene_dict = {
@@ -135,7 +136,7 @@ def test_spark_df_source(self):
135136
self.assertTrue(len(path_df.columns) == 6) # three bands times {path, tile}
136137
self.assertTrue(path_df.select('b1_path').distinct().count() == 3) # as per scene_dict
137138
b1_paths_maybe = path_df.select('b1_path').distinct().collect()
138-
b1_paths = [s.format('1') for s in scene_dict.values()]
139+
b1_paths = [self.path(s, 1) for s in [1, 2, 3]]
139140
self.assertTrue(all([row.b1_path in b1_paths for row in b1_paths_maybe]))
140141

141142
def test_pandas_source(self):

0 commit comments

Comments
 (0)