test: add concat function

substrait-io · May 4, 2023 · 4557c0f · 4557c0f
1 parent a8fddcb
commit 4557c0f
Show file tree

Hide file tree

Showing 6 changed files with 56 additions and 5 deletions.
diff --git a/bft/testers/postgres/runner.py b/bft/testers/postgres/runner.py
@@ -59,7 +59,7 @@ def run_sql_case(self, case: Case, mapping: SqlMapping) -> SqlCaseResult:
             arg_vals = ",".join(
                 [
                     "'" + literal_to_str(arg) + "'"
-                    if arg.type == "string"
+                    if arg.type == "string" and arg.value is not None
                     else literal_to_str(arg)
                     for arg in case.args
                 ]

diff --git a/bft/testers/sqlite/runner.py b/bft/testers/sqlite/runner.py
@@ -50,7 +50,7 @@ def run_sql_case(self, case: Case, mapping: SqlMapping) -> SqlCaseResult:
             arg_vals = ",".join(
                 [
                     "'" + literal_to_str(arg) + "'"
-                    if arg.type == "string"
+                    if arg.type == "string" and arg.value is not None
                     else literal_to_str(arg)
                     for arg in case.args
                 ]

diff --git a/cases/string/concat.yaml b/cases/string/concat.yaml
@@ -0,0 +1,42 @@
+function: concat
+cases:
+  - group:
+      id: basic
+      description: Basic examples without any special cases
+    args:
+      - value: 'abcd'
+        type: string
+      - value: 'efg'
+        type: string
+    result:
+      value: 'abcdefg'
+      type: string
+  - group:
+      id: null_input
+      description: Examples with null as input
+    args:
+      - value: 'abcd'
+        type: string
+      - value: null
+        type: string
+    result:
+      value: null
+      type: string
+  - group: null_input
+    args:
+      - value: null
+        type: string
+      - value: 'abcd'
+        type: string
+    result:
+      value: null
+      type: string
+  - group: null_input
+    args:
+      - value: null
+        type: string
+      - value: null
+        type: string
+    result:
+      value: null
+      type: string
diff --git a/dialects/postgres.yaml b/dialects/postgres.yaml
@@ -56,4 +56,7 @@ scalar_functions:
     infix: False
   - name: substring
     local_name: substring
-    infix: False
+    infix: False
+  - name: concat
+    local_name: "||"
+    infix: True
diff --git a/dialects/sqlite.yaml b/dialects/sqlite.yaml
@@ -40,4 +40,7 @@ scalar_functions:
     infix: False
   - name: substring
     local_name: substring
-    infix: False
+    infix: False
+  - name: concat
+    local_name: "||"
+    infix: True
diff --git a/dialects/velox_presto.yaml b/dialects/velox_presto.yaml
@@ -64,4 +64,7 @@ scalar_functions:
     infix: False
   - name: substring
     local_name: substring
-    infix: False
+    infix: False
+  - name: concat
+    local_name: concat
+    infix: False