fix test failures

apache · Aug 27, 2024 · f17b3d0 · f17b3d0
1 parent 7aa7eae
commit f17b3d0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/spark/common/src/test/java/org/apache/sedona/core/spatialOperator/JoinQueryTest.java b/spark/common/src/test/java/org/apache/sedona/core/spatialOperator/JoinQueryTest.java
@@ -19,10 +19,7 @@
 package org.apache.sedona.core.spatialOperator;
 
 import java.io.IOException;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 import org.apache.sedona.core.enums.GridType;
 import org.apache.sedona.core.enums.IndexType;
@@ -53,7 +50,9 @@ public static void teardown() {
 
   @Parameterized.Parameters(name = "JoinQueryTest-{index}: {0}")
   public static SpatialPredicate[] spatialPredicates() {
-    return SpatialPredicate.values();
+    return Arrays.stream(SpatialPredicate.values())
+        .filter(predicate -> predicate != SpatialPredicate.KNN)
+        .toArray(SpatialPredicate[]::new);
   }
 
   private final SpatialPredicate spatialPredicate;

diff --git a/spark/common/src/test/java/org/apache/sedona/core/spatialOperator/RangeQueryTest.java b/spark/common/src/test/java/org/apache/sedona/core/spatialOperator/RangeQueryTest.java
@@ -19,6 +19,7 @@
 package org.apache.sedona.core.spatialOperator;
 
 import java.io.IOException;
+import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
@@ -50,7 +51,9 @@ public static void teardown() {
 
   @Parameterized.Parameters(name = "RangeQueryTest-{index}: {0}")
   public static SpatialPredicate[] spatialPredicates() {
-    return SpatialPredicate.values();
+    return Arrays.stream(SpatialPredicate.values())
+        .filter(predicate -> predicate != SpatialPredicate.KNN)
+        .toArray(SpatialPredicate[]::new);
   }
 
   private final SpatialPredicate spatialPredicate;