fix test

snowflakedb · Mar 8, 2024 · 9934af0 · 9934af0
1 parent a6d266f
commit 9934af0
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/src/test/java/com/snowflake/snowpark_test/JavaGeographySuite.java b/src/test/java/com/snowflake/snowpark_test/JavaGeographySuite.java
@@ -17,7 +17,7 @@ public void testRoundTrip() {
     assert (Geography.fromGeoJSON(testData).asGeoJSON().equals(testData));
     assert (Geography.fromGeoJSON(testData).toString().equals(testData));
 
-    assert (Geometry.fromGeoJSON(testData2).toString().equals(testData));
+    assert (Geometry.fromGeoJSON(testData2).toString().equals(testData2));
   }
 
   @Test

diff --git a/src/test/scala/com/snowflake/snowpark_test/IndependentClassSuite.scala b/src/test/scala/com/snowflake/snowpark_test/IndependentClassSuite.scala
@@ -15,7 +15,10 @@ class IndependentClassSuite extends FunSuite {
 
     checkDependencies(
       "target/classes/com/snowflake/snowpark/types/Variant$.class",
-      Seq("com.snowflake.snowpark.types.Variant", "com.snowflake.snowpark.types.Geography"))
+      Seq(
+        "com.snowflake.snowpark.types.Variant",
+        "com.snowflake.snowpark.types.Geography",
+        "com.snowflake.snowpark.types.Geometry"))
   }
 
   test("java variant") {
@@ -42,6 +45,22 @@ class IndependentClassSuite extends FunSuite {
       Seq("com.snowflake.snowpark_java.types.Geography"))
   }
 
+  test("scala geometry") {
+    checkDependencies(
+      "target/classes/com/snowflake/snowpark/types/Geometry.class",
+      Seq("com.snowflake.snowpark.types.Geometry"))
+
+    checkDependencies(
+      "target/classes/com/snowflake/snowpark/types/Geometry$.class",
+      Seq("com.snowflake.snowpark.types.Geometry"))
+  }
+
+  test("java geometry") {
+    checkDependencies(
+      "target/classes/com/snowflake/snowpark_java/types/Geometry.class",
+      Seq("com.snowflake.snowpark_java.types.Geometry"))
+  }
+
   // negative test, to make sure this test method works
   test("session") {
     assertThrows[TestFailedException] {