Fix raster tests in R.

databrickslabs · Feb 13, 2024 · 32d6f76 · 32d6f76
1 parent 6a79482
commit 32d6f76
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/R/sparkR-mosaic/sparkrMosaic/tests/testthat/testRasterFunctions.R b/R/sparkR-mosaic/sparkrMosaic/tests/testthat/testRasterFunctions.R
@@ -61,13 +61,13 @@ test_that("raster flatmap functions behave as intended", {
   tessellate_sdf <- withColumn(tessellate_sdf, "rst_tessellate", rst_tessellate(column("tile"), lit(3L)))
 
   expect_no_error(write.df(tessellate_sdf, source = "noop", mode = "overwrite"))
-  expect_equal(nrow(tessellate_sdf), 140)
+  expect_equal(nrow(tessellate_sdf), 66)
 
   overlap_sdf <- generate_singleband_raster_df()
   overlap_sdf <- withColumn(overlap_sdf, "rst_to_overlapping_tiles", rst_to_overlapping_tiles(column("tile"), lit(200L), lit(200L), lit(10L)))
 
   expect_no_error(write.df(overlap_sdf, source = "noop", mode = "overwrite"))
-  expect_equal(nrow(overlap_sdf), 196)
+  expect_equal(nrow(overlap_sdf), 87)
 })
 
 test_that("raster aggregation functions behave as intended", {

diff --git a/R/sparklyr-mosaic/sparklyrMosaic/tests/testthat/testRasterFunctions.R b/R/sparklyr-mosaic/sparklyrMosaic/tests/testthat/testRasterFunctions.R
@@ -90,13 +90,13 @@ test_that("raster flatmap functions behave as intended", {
     mutate(rst_tessellate = rst_tessellate(tile, 3L))
 
   expect_no_error(spark_write_source(tessellate_sdf, "noop", mode = "overwrite"))
-  expect_equal(sdf_nrow(tessellate_sdf), 140)
+  expect_equal(sdf_nrow(tessellate_sdf), 66)
 
   overlap_sdf <- generate_singleband_raster_df() %>%
     mutate(rst_to_overlapping_tiles = rst_to_overlapping_tiles(tile, 200L, 200L, 10L))
 
   expect_no_error(spark_write_source(overlap_sdf, "noop", mode = "overwrite"))
-  expect_equal(sdf_nrow(overlap_sdf), 196)
+  expect_equal(sdf_nrow(overlap_sdf), 87)
 
 })