4/7 gen tests

david518yang · Apr 29, 2024 · 6883a78 · 6883a78
1 parent afb2994
commit 6883a78
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 8 deletions.
diff --git a/src/core.js b/src/core.js
@@ -84,7 +84,7 @@ export function returnStatement(expression) {
 }
 
 export function shortReturnStatement() {
-    return { kind: 'shortreturnStatement' }
+    return { kind: 'shortReturnStatement' }
 }
 
 export function ifStatement(test, consequent, alternate) {

diff --git a/src/generator.js b/src/generator.js
@@ -116,15 +116,22 @@ export default function generate(program) {
       s.body.forEach(gen)
       output.push("}")
     },
-    // Conditional(e) {
-    //   return `((${gen(e.test)}) ? (${gen(e.consequent)}) : (${gen(e.alternate)}))`
-    // },
+    Conditional(e) {
+      return `((${gen(e.test)}) ? (${gen(e.consequent)}) : (${gen(e.alternate)}))`
+    },
     BinaryExpression(e) {
       const op = { "==": "===", "!=": "!==" }[e.op] ?? e.op
       return `(${gen(e.left)} ${op} ${gen(e.right)})`
     },
-    NegationExpression(e) {
+    UnaryExpression(e) {
       const operand = gen(e.operand)
+      if (e.op === "some") {
+        return operand
+      } else if (e.op === "#") {
+        return `${operand}.length`
+      } else if (e.op === "random") {
+        return `((a=>a[~~(Math.random()*a.length)])(${operand}))`
+      }
       return `${e.op}(${operand})`
     },
     SubscriptExpression(e) {

diff --git a/test/generator.test.js b/test/generator.test.js
@@ -19,8 +19,8 @@ const fixtures = [
     `,
     expected: dedent`
       let x_1 = 21;
-      x_1 = x_1 + 1;
-      x_1 = x_1 - 1;
+      x_1 = (x_1 + 1);
+      x_1 = (x_1 - 1);
       let y_2 = true;
     `,
   },
@@ -97,7 +97,6 @@ const fixtures = [
       func g(): bool {
         return false;
       }
-      f(z, g());
     `,
     expected: dedent`
       let z_1 = 0.5;