add test for fromSubstrait with a plan for "select true"

scgkiran · Dec 18, 2024 · f541de0 · f541de0
1 parent 0d85c4f
commit f541de0
Showing 1 changed file with 62 additions and 0 deletions.
diff --git a/test/c/test_substrait_c_api.cpp b/test/c/test_substrait_c_api.cpp
@@ -320,3 +320,65 @@ TEST_CASE("Test C VirtualTable input Expression", "[substrait-api]") {
   REQUIRE(CHECK_COLUMN(result, 0, {2, 6}));
   REQUIRE(CHECK_COLUMN(result, 1, {4, 8}));
 }
+
+
+TEST_CASE("Test C Simple select true", "[substrait-api]") {
+	DuckDB db(nullptr);
+	Connection con(db);
+
+	auto json = R"({
+    "version": {
+        "minorNumber": 29,
+        "producer": "substrait-go"
+    },
+    "relations": [
+        {
+            "root": {
+                "input": {
+                    "read": {
+                        "common": {
+                            "direct": {}
+                        },
+                        "baseSchema": {
+                            "names": [
+                                "C1"
+                            ],
+                            "struct": {
+                                "types": [
+                                    {
+                                        "bool": {
+                                            "nullability": "NULLABILITY_NULLABLE"
+                                        }
+                                    }
+                                ],
+                                "nullability": "NULLABILITY_REQUIRED"
+                            }
+                        },
+                        "virtualTable": {
+                            "expressions": [
+                                {
+                                    "fields": [
+                                        {
+                                            "literal": {
+                                                "boolean": true,
+                                                "nullable": true
+                                            }
+                                        }
+                                    ]
+                                }
+                            ]
+                        }
+                    }
+                },
+                "names": [
+                    "C1"
+                ]
+            }
+        }
+    ]
+})";
+
+
+	auto result = con.FromSubstraitJSON(json);
+	REQUIRE(CHECK_COLUMN(result, 0, {true}));
+}