From d362ef26db894ea52c6abd34489aa9efdbc0c89b Mon Sep 17 00:00:00 2001
From: Akshay Nair <phenax5@gmail.com>
Date: Sat, 14 Jan 2023 13:22:39 +0530
Subject: chore: adds test cases + fixes Test effect scope

---
 src/eval-env/builtins.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/eval-env/builtins.ts')

diff --git a/src/eval-env/builtins.ts b/src/eval-env/builtins.ts
index 669a005..35f8485 100644
--- a/src/eval-env/builtins.ts
+++ b/src/eval-env/builtins.ts
@@ -6,10 +6,10 @@ const applyFunc = (ctx: Ctx, fn: Type | undefined, val: string): Type => {
   const resultType = (() => {
     const baseTypes = fn
       ?.getBaseTypes()
-      .flatMap((t) => t.getSymbol()?.getName())
+      .flatMap((t) => t.getSymbol()?.getName() ?? [])
 
-    if (baseTypes?.includes('Kind1')) {
-      const [_, resultNode] = ctx.createResult(
+    if (!!fn?.getProperty('return') || baseTypes?.includes('Kind1')) {
+      const [_key, resultNode] = ctx.createResult(
         `(${ctx.typeToString(fn)} & { input: ${val} })['return']`
       )
       return resultNode
@@ -53,7 +53,7 @@ const applyFunc = (ctx: Ctx, fn: Type | undefined, val: string): Type => {
   // TODO: Cleanup unwanted result node values
 
   if (!resultType) {
-    throw new Error('Fuck shit')
+    throw new Error('Couldnt get result for function application')
   }
 
   return resultType
-- 
cgit v1.3.1