add test

snowflakedb · Nov 14, 2023 · c05b0c5 · c05b0c5
1 parent 02e497c
commit c05b0c5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/test/java/com/snowflake/snowpark_test/JavaUDTFSuite.java b/src/test/java/com/snowflake/snowpark_test/JavaUDTFSuite.java
@@ -345,6 +345,12 @@ public void partitionBy() {
 
     DataFrame result4 = df.join(tf, map, new Column[] {}, new Column[] {});
     result4.show();
+
+    DataFrame result5 = df.join(tf.call(map), new Column[] {df.col("a")}, new Column[] {df.col("b")});
+    checkAnswer(
+            result5,
+            new Row[] {Row.create("a", null, "{b=2, c=1}"), Row.create("d", null, "{e=1}")},
+            true);
   }
 }