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 <[email protected]> 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)
