test generator harnesskai/test-generator

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/type-consistency/spec/function.ts b/tests/type-consistency/spec/function.ts
new file mode 100644
index 0000000..ffd069d
--- /dev/null
+++ b/tests/type-consistency/spec/function.ts
@@ -0,0 +1,26 @@
+import path from "path";
+import { createTestHarness } from "../harness";
+
+export default createTestHarness({
+  harnessName: "Function",
+  generatedPath: path.join(__dirname, "..", "generated"),
+})
+  .createFunctionTest({
+    name: "f(x) = x",
+    program: "fn(x, x)",
+    cases: [
+      { input: "hello", output: "hello" },
+      { input: 1, output: 1 },
+      { input: [1, 2, 3], output: [1, 2, 3] },
+      { input: true, output: true },
+    ],
+  })
+  .createFunctionTest({
+    name: "f(x) = x + 5",
+    program: "fn(x, add(x,5))",
+    cases: [
+      { input: 0, output: 5 },
+      { input: 5, output: 10 },
+      { input: 500, output: 505 },
+    ],
+  });