fix: recursive lambda functions fixed

mrunix00 · Apr 16, 2024 · 6b55493 · 6b55493
1 parent eafd9b1
commit 6b55493
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/bytecode/builtin_functions/DefineFunction.cpp b/src/bytecode/builtin_functions/DefineFunction.cpp
@@ -8,9 +8,9 @@ namespace Bytecode::BuiltinFunctions {
  std::vector<Instruction *> &instructions,
  Segment *result) {
  if (args[0]->type == SyntaxTreeNode::TokenNode) {
- args[1]->compile(result, program, instructions);
  const auto reg = program.declare_global(
  ((TokenNode *) args[0])->getName());
+ args[1]->compile(result, program, instructions);
  instructions.push_back(new StoreGlobal(reg));
  } else if (args[0]->type == SyntaxTreeNode::Expression) {
  auto segment = new Segment({});

diff --git a/tests/lambda_test.cpp b/tests/lambda_test.cpp
@@ -33,3 +33,15 @@ TEST(lambda_test, ShouldPassLambdaAsArgument) {
 
  EXPECT_EQ(actual_result, expected_result);
 }
+
+TEST(lambda_test, RecursiveLambdaFunction) {
+ const auto expression =
+ "(define fib (lambda (n)"
+ "(if (< n 2) n (+ (fib (- n 1)) (fib (- n 2))))))"
+ "(fib 10)";
+
+ const auto expected_result = StackObject((double) 55);
+ const auto actual_result = run_program(expression);
+
+ EXPECT_EQ(actual_result, expected_result);
+}