@@ -43,11 +43,11 @@ func TestBooleanExpressions(t *testing.T) {
4343
4444func TestConditionalExpressions (t * testing.T ) {
4545 tests := []vmTestCase {
46- {input : "if (true) { 10 }" , expected : 10 },
47- {input : "if (true) { 10 } else { 20 }" , expected : 10 },
48- {input : "if (false) { 10 } else { 20 }" , expected : 20 },
49- {input : "if (1 > 10) { 10 } else { 20 }" , expected : 20 },
50- {input : "if (1 > 10) { 10 } else if(true) { 400 } else { 20 }" , expected : 400 },
46+ {input : "if (true) { return 10 }" , expected : 10 },
47+ {input : "if (true) { return 10 } else { 20 }" , expected : 10 },
48+ {input : "if (false) { return 10 } else { return 20 }" , expected : 20 },
49+ {input : "if (1 > 10) { return 10 } else { return 20 }" , expected : 20 },
50+ {input : "if (1 > 10) { return 10 } else if(true) { return 400 } else { return 20 }" , expected : 400 },
5151 {input : "if (1 > 10) {}" , expected : Null },
5252 {input : "if (false) {}" , expected : Null },
5353 }
@@ -174,7 +174,7 @@ func TestVM_CallExpressionsWrongArguments(t *testing.T) {
174174 t .Fatalf ("compiler error: %s" , err )
175175 }
176176 vm := Create (comp .Bytecode ())
177- err = vm .Run ()
177+ err = vm .Run (nil )
178178 if err == nil {
179179 t .Fatalf ("expected VM error but resulted in none." )
180180 }
@@ -345,7 +345,7 @@ func runVmTests(t *testing.T, tests []vmTestCase) {
345345 }
346346
347347 vm := Create (comp .Bytecode ())
348- err = vm .Run ()
348+ err = vm .Run (nil )
349349 if err != nil {
350350 t .Fatalf ("vm error: %s" , err )
351351 }
0 commit comments