more test for boolean expression & fuzzer

3cham · May 12, 2024 · 2ca0ea6 · 2ca0ea6
1 parent 49279df
commit 2ca0ea6
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/fuzzer/src/main/kotlin/Fuzzer.kt b/fuzzer/src/main/kotlin/Fuzzer.kt
@@ -33,7 +33,7 @@ class Fuzzer {
         return createRecordBatch(schema, columns)
     }
 
-    private fun createRecordBatch(schema: Schema, column: List<List<Any?>>): RecordBatch {
+    fun createRecordBatch(schema: Schema, column: List<List<Any?>>): RecordBatch {
         val arrowVectors = schema.fields.withIndex().map { field ->
             val fv = FieldVectorFactory.create(field.value.dataType, column[field.index].size)
 

diff --git a/physical-plan/src/test/kotlin/BooleanExpressionTest.kt b/physical-plan/src/test/kotlin/BooleanExpressionTest.kt
@@ -3,6 +3,7 @@ import io.hqew.kquery.datatypes.Field
 import io.hqew.kquery.datatypes.Schema
 import io.hqew.kquery.fuzzer.Fuzzer
 import io.hqew.kquery.physical.expressions.ColumnExpression
+import io.hqew.kquery.physical.expressions.EqExpression
 import io.hqew.kquery.physical.expressions.LtEqExpression
 import org.junit.Test
 import org.junit.jupiter.api.TestInstance
@@ -30,4 +31,29 @@ class BooleanExpressionTest {
             assertEquals(expected, lteq.getValue(it))
         }
     }
+
+    @Test
+    fun `test EqExpression with string`() {
+        val schema = Schema(listOf(
+            Field("0", ArrowTypes.StringType),
+            Field("1", ArrowTypes.StringType)
+        ))
+
+        val l = ColumnExpression(0)
+        val r = ColumnExpression(1)
+
+        val values = listOf(
+            listOf("aaa", "bbb", "ccc"),
+            listOf("aaa", "bbb", "ccd")
+        )
+
+        val input = Fuzzer().createRecordBatch(schema, values)
+
+        val eq = EqExpression(l, r).evaluate(input)
+
+        (0 until eq.size()).forEach {
+            val expected = input.fields[0].getValue(it) == input.fields[1].getValue(it)
+            assertEquals(expected, eq.getValue(it))
+        }
+    }
 }