[sedona] 08/09: Fix more test cases

[ji...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

jiayu pushed a commit to branch fix-lonlat-order
in repository https://gitbox.apache.org/repos/asf/sedona.git

commit 86ec83f3f108e171ca03ad4285ee9d3ab767bd44
Author: Jia Yu <ji...@apache.org>
AuthorDate: Tue Aug 8 12:59:57 2023 -0700

    Fix more test cases
---
 .../test/java/org/apache/sedona/viz/ChoroplethmapTest.java |  2 ++
 .../test/scala/org/apache/sedona/viz/rdd/scalaTest.scala   | 14 +++++++-------
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/viz/src/test/java/org/apache/sedona/viz/ChoroplethmapTest.java b/viz/src/test/java/org/apache/sedona/viz/ChoroplethmapTest.java
index eb176417..ab932642 100644
--- a/viz/src/test/java/org/apache/sedona/viz/ChoroplethmapTest.java
+++ b/viz/src/test/java/org/apache/sedona/viz/ChoroplethmapTest.java
@@ -55,6 +55,7 @@ public class ChoroplethmapTest
     {
         PointRDD spatialRDD = new PointRDD(sparkContext, PointInputLocation, PointOffset, PointSplitter, false, PointNumPartitions);
         RectangleRDD queryRDD = new RectangleRDD(sparkContext, RectangleInputLocation, RectangleSplitter, false, RectangleNumPartitions);
+        spatialRDD.analyze();
         spatialRDD.spatialPartitioning(GridType.KDBTREE);
         queryRDD.spatialPartitioning(spatialRDD.getPartitioner());
         spatialRDD.buildIndex(IndexType.RTREE, true);
@@ -87,6 +88,7 @@ public class ChoroplethmapTest
         //UserSuppliedPolygonMapper userSuppliedPolygonMapper = new UserSuppliedPolygonMapper();
         PointRDD spatialRDD = new PointRDD(sparkContext, PointInputLocation, PointOffset, PointSplitter, false, PointNumPartitions);
         PolygonRDD queryRDD = new PolygonRDD(sparkContext, PolygonInputLocation, PolygonSplitter, false, PolygonNumPartitions);
+        spatialRDD.analyze();
         spatialRDD.spatialPartitioning(GridType.KDBTREE);
         queryRDD.spatialPartitioning(spatialRDD.getPartitioner());
         spatialRDD.buildIndex(IndexType.RTREE, true);
diff --git a/viz/src/test/scala/org/apache/sedona/viz/rdd/scalaTest.scala b/viz/src/test/scala/org/apache/sedona/viz/rdd/scalaTest.scala
index e417718a..7d40e13f 100644
--- a/viz/src/test/scala/org/apache/sedona/viz/rdd/scalaTest.scala
+++ b/viz/src/test/scala/org/apache/sedona/viz/rdd/scalaTest.scala
@@ -18,11 +18,6 @@
  */
 package org.apache.sedona.viz.rdd
 
-import java.awt.Color // scalastyle:ignore illegal.imports
-import java.io.FileInputStream
-import java.util.Properties
-
-import org.locationtech.jts.geom.Envelope
 import org.apache.log4j.{Level, Logger}
 import org.apache.sedona.common.enums.FileDataSplitter
 import org.apache.sedona.core.enums.{GridType, IndexType}
@@ -30,14 +25,18 @@ import org.apache.sedona.core.formatMapper.EarthdataHDFPointMapper
 import org.apache.sedona.core.spatialOperator.JoinQuery
 import org.apache.sedona.core.spatialRDD.{PointRDD, PolygonRDD, RectangleRDD}
 import org.apache.sedona.viz.`extension`.visualizationEffect.{ChoroplethMap, HeatMap, ScatterPlot}
-import org.apache.sedona.viz.core.{ImageGenerator, RasterOverlayOperator}
 import org.apache.sedona.viz.core.Serde.SedonaVizKryoRegistrator
+import org.apache.sedona.viz.core.{ImageGenerator, RasterOverlayOperator}
 import org.apache.sedona.viz.utils.{ColorizeOption, ImageType}
 import org.apache.spark.serializer.KryoSerializer
-import org.apache.spark.storage.StorageLevel
 import org.apache.spark.{SparkConf, SparkContext}
+import org.locationtech.jts.geom.Envelope
 import org.scalatest.{BeforeAndAfterAll, FunSpec}
 
+import java.awt.Color
+import java.io.FileInputStream
+import java.util.Properties
+
 class scalaTest extends FunSpec with BeforeAndAfterAll{
   val sparkConf = new SparkConf().setAppName("scalaTest").setMaster("local[*]")
   sparkConf.set("spark.serializer", classOf[KryoSerializer].getName)
@@ -130,6 +129,7 @@ class scalaTest extends FunSpec with BeforeAndAfterAll{
     it("should pass choropleth map") {
       val spatialRDD = new PointRDD(sparkContext, PointInputLocation, PointOffset, PointSplitter, false, PointNumPartitions)
       val queryRDD = new PolygonRDD(sparkContext, PolygonInputLocation, PolygonSplitter, false, PolygonNumPartitions)
+      spatialRDD.analyze();
       spatialRDD.spatialPartitioning(GridType.KDBTREE)
       queryRDD.spatialPartitioning(spatialRDD.getPartitioner)
       spatialRDD.buildIndex(IndexType.RTREE, true)