From c9dd1ce1b9b1867a9eaaed4d53dcda51e5ed003b Mon Sep 17 00:00:00 2001 From: Furqaanahmed Khan Date: Thu, 7 Dec 2023 20:31:45 -0500 Subject: [PATCH] remove if-else block and print statement --- .../sedona_sql/expressions/raster/AggregateFunctions.scala | 5 +---- .../test/scala/org/apache/sedona/sql/rasteralgebraTest.scala | 2 -- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/spark/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/raster/AggregateFunctions.scala b/spark/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/raster/AggregateFunctions.scala index 8063c94d41..1fa1cb6e7a 100644 --- a/spark/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/raster/AggregateFunctions.scala +++ b/spark/common/src/main/scala/org/apache/spark/sql/sedona_sql/expressions/raster/AggregateFunctions.scala @@ -63,10 +63,7 @@ class RS_Union_Aggr extends Aggregator[(GridCoverage2D, Int), ArrayBuffer[BandDa val widthNewRaster = RasterAccessors.getWidth(raster) val heightNewRaster = RasterAccessors.getHeight(raster) - if (width == widthNewRaster && height == heightNewRaster) - true - else - false + width == widthNewRaster && height == heightNewRaster } } diff --git a/spark/common/src/test/scala/org/apache/sedona/sql/rasteralgebraTest.scala b/spark/common/src/test/scala/org/apache/sedona/sql/rasteralgebraTest.scala index b386cb745a..ffd179ae35 100644 --- a/spark/common/src/test/scala/org/apache/sedona/sql/rasteralgebraTest.scala +++ b/spark/common/src/test/scala/org/apache/sedona/sql/rasteralgebraTest.scala @@ -890,8 +890,6 @@ class rasteralgebraTest extends TestBaseScala with BeforeAndAfter with GivenWhen df = df.selectExpr("RS_Union_aggr(raster, index) as rasters") - println(df.show(10, 50)) - val actualBands = df.selectExpr("RS_NumBands(rasters)").first().get(0) val expectedBands = 4 assertEquals(expectedBands, actualBands)