diff options
Diffstat (limited to 'interp')
| -rw-r--r-- | interp/interpreter_test.go | 209 |
1 files changed, 107 insertions, 102 deletions
diff --git a/interp/interpreter_test.go b/interp/interpreter_test.go index bc353bd..28a80d4 100644 --- a/interp/interpreter_test.go +++ b/interp/interpreter_test.go @@ -3,6 +3,7 @@ package interp_test import ( "fmt" "log" + "strings" "testing" "github.com/mvertes/parscan/interp" @@ -31,7 +32,7 @@ func gen(test etest) func(*testing.T) { if e != nil { errStr = e.Error() } - if errStr != test.err { + if !strings.Contains(errStr, test.err) { t.Errorf("got error %#v, want error %#v", errStr, test.err) } if res := fmt.Sprintf("%v", r); test.err == "" && res != test.res { @@ -48,75 +49,81 @@ func run(t *testing.T, tests []etest) { func TestExpr(t *testing.T) { run(t, []etest{ - {src: "", res: "<invalid reflect.Value>"}, - {src: "1+2", res: "3"}, - {src: "1+", err: "block not terminated"}, - {src: "a := 1 + 2; b := 0; a + 1", res: "4"}, - {src: "1+(2+3)", res: "6"}, - {src: "(1+2)+3", res: "6"}, - {src: "(6+(1+2)+3)+5", res: "17"}, - {src: "(6+(1+2+3)+5", err: "1:1: block not terminated"}, - {src: "a := 2; a = 3; a", res: "3"}, - {src: "2 * 3 + 1 == 7", res: "true"}, - {src: "7 == 2 * 3 + 1", res: "true"}, - {src: "1 + 3 * 2 == 2 * 3 + 1", res: "true"}, - {src: "a := 1 + 3 * 2 == 2 * 3 + 1; a", res: "true"}, - {src: "-2", res: "-2"}, - {src: "-2 + 5", res: "3"}, - {src: "5 + -2", res: "3"}, - {src: "!false", res: "true"}, - {src: `a := "hello"`, res: "hello"}, + {src: "", res: "<invalid reflect.Value>"}, // #00 + {src: "1+2", res: "3"}, // #01 + {src: "1+", err: "block not terminated"}, // #02 + {src: "a := 1 + 2; b := 0; a + 1", res: "4"}, // #03 + {src: "1+(2+3)", res: "6"}, // #04 + {src: "(1+2)+3", res: "6"}, // #05 + {src: "(6+(1+2)+3)+5", res: "17"}, // #06 + {src: "(6+(1+2+3)+5", err: "1:1: block not terminated"}, // #07 + {src: "a := 2; a = 3; a", res: "3"}, // #08 + {src: "2 * 3 + 1 == 7", res: "true"}, // #09 + {src: "7 == 2 * 3 + 1", res: "true"}, // #10 + {src: "1 + 3 * 2 == 2 * 3 + 1", res: "true"}, // #11 + {src: "a := 1 + 3 * 2 == 2 * 3 + 1; a", res: "true"}, // #12 + {src: "-2", res: "-2"}, // #13 + {src: "-2 + 5", res: "3"}, // #14 + {src: "5 + -2", res: "3"}, // #15 + {src: "!false", res: "true"}, // #16 + {src: `a := "hello"`, res: "hello"}, // #17 + }) +} + +func TestCompare(t *testing.T) { + run(t, []etest{ + {src: "a := 1; a < 2", res: "true"}, }) } func TestLogical(t *testing.T) { run(t, []etest{ - {src: "true && false", res: "false"}, - {src: "true && true", res: "true"}, - {src: "true && true && false", res: "false"}, - {src: "false || true && true", res: "true"}, - {src: "2 < 3 && 1 > 2 || 3 == 3", res: "true"}, - {src: "2 > 3 && 1 > 2 || 3 == 3", res: "true"}, - {src: "2 > 3 || 2 == 1+1 && 3>0", res: "true"}, - {src: "2 > 3 || 2 == 1+1 && 3>4 || 1<2", res: "true"}, - {src: "a := 1+1 < 3 && 4 == 2+2; a", res: "true"}, - {src: "a := 1+1 < 3 || 3 == 2+2; a", res: "true"}, + {src: "true && false", res: "false"}, // #00 + {src: "true && true", res: "true"}, // #01 + {src: "true && true && false", res: "false"}, // #02 + {src: "false || true && true", res: "true"}, // #03 + {src: "2 < 3 && 1 > 2 || 3 == 3", res: "true"}, // #04 + {src: "2 > 3 && 1 > 2 || 3 == 3", res: "true"}, // #05 + {src: "2 > 3 || 2 == 1+1 && 3>0", res: "true"}, // #06 + {src: "2 > 3 || 2 == 1+1 && 3>4 || 1<2", res: "true"}, // #07 + {src: "a := 1+1 < 3 && 4 == 2+2; a", res: "true"}, // #08 + {src: "a := 1+1 < 3 || 3 == 2+2; a", res: "true"}, // #09 }) } func TestFunc(t *testing.T) { run(t, []etest{ - {src: "func f() int {return 2}; a := f(); a", res: "2"}, - {src: "func f() int {return 2}; f()", res: "2"}, - {src: "func f(a int) int {return a+2}; f(3)", res: "5"}, - {src: "func f(a int) int {if a < 4 {a = 5}; return a}; f(3)", res: "5"}, - {src: "func f(a int) int {return a+2}; 7 - f(3)", res: "2"}, - {src: "func f(a int) int {return a+2}; f(5) - f(3)", res: "2"}, - {src: "func f(a int) int {return a+2}; f(3) - 2", res: "3"}, - {src: "func f(a, b, c int) int {return a+b-c} ; f(7, 1, 3)", res: "5"}, - {src: "var a int; func f() {a = a+2}; f(); a", res: "2"}, - {src: "var f = func(a int) int {return a+3}; f(2)", res: "5"}, + {src: "func f() int {return 2}; a := f(); a", res: "2"}, // #00 + {src: "func f() int {return 2}; f()", res: "2"}, // #01 + {src: "func f(a int) int {return a+2}; f(3)", res: "5"}, // #02 + {src: "func f(a int) int {if a < 4 {a = 5}; return a}; f(3)", res: "5"}, // #03 + {src: "func f(a int) int {return a+2}; 7 - f(3)", res: "2"}, // #04 + {src: "func f(a int) int {return a+2}; f(5) - f(3)", res: "2"}, // #05 + {src: "func f(a int) int {return a+2}; f(3) - 2", res: "3"}, // #06 + {src: "func f(a, b, c int) int {return a+b-c} ; f(7, 1, 3)", res: "5"}, // #07 + {src: "var a int; func f() {a = a+2}; f(); a", res: "2"}, // #08 + {src: "var f = func(a int) int {return a+3}; f(2)", res: "5"}, // #09 }) } func TestIf(t *testing.T) { run(t, []etest{ - {src: "a := 0; if a == 0 { a = 2 } else { a = 1 }; a", res: "2"}, - {src: "a := 0; if a == 1 { a = 2 } else { a = 1 }; a", res: "1"}, - {src: "a := 0; if a == 1 { a = 2 } else if a == 0 { a = 3 } else { a = 1 }; a", res: "3"}, - {src: "a := 0; if a == 1 { a = 2 } else if a == 2 { a = 3 } else { a = 1 }; a", res: "1"}, - {src: "a := 1; if a > 0 && a < 2 { a = 3 }; a", res: "3"}, - {src: "a := 1; if a < 0 || a < 2 { a = 3 }; a", res: "3"}, + {src: "a := 0; if a == 0 { a = 2 } else { a = 1 }; a", res: "2"}, // #00 + {src: "a := 0; if a == 1 { a = 2 } else { a = 1 }; a", res: "1"}, // #01 + {src: "a := 0; if a == 1 { a = 2 } else if a == 0 { a = 3 } else { a = 1 }; a", res: "3"}, // #02 + {src: "a := 0; if a == 1 { a = 2 } else if a == 2 { a = 3 } else { a = 1 }; a", res: "1"}, // #03 + {src: "a := 1; if a > 0 && a < 2 { a = 3 }; a", res: "3"}, // #04 + {src: "a := 1; if a < 0 || a < 2 { a = 3 }; a", res: "3"}, // #05 }) } func TestFor(t *testing.T) { run(t, []etest{ - {src: "a := 0; for i := 0; i < 3; i = i+1 {a = a+i}; a", res: "3"}, - {src: "func f() int {a := 0; for i := 0; i < 3; i = i+1 {a = a+i}; return a}; f()", res: "3"}, - {src: "a := 0; for {a = a+1; if a == 3 {break}}; a", res: "3"}, - {src: "func f() int {a := 0; for {a = a+1; if a == 3 {break}}; return a}; f()", res: "3"}, - {src: "func f() int {a := 0; for {a = a+1; if a < 3 {continue}; break}; return a}; f()", res: "3"}, + {src: "a := 0; for i := 0; i < 3; i = i+1 {a = a+i}; a", res: "3"}, // #00 + {src: "func f() int {a := 0; for i := 0; i < 3; i = i+1 {a = a+i}; return a}; f()", res: "3"}, // #01 + {src: "a := 0; for {a = a+1; if a == 3 {break}}; a", res: "3"}, // #02 + {src: "func f() int {a := 0; for {a = a+1; if a == 3 {break}}; return a}; f()", res: "3"}, // #03 + {src: "func f() int {a := 0; for {a = a+1; if a < 3 {continue}; break}; return a}; f()", res: "3"}, // #04 }) } @@ -155,15 +162,15 @@ func TestSwitch(t *testing.T) { } ` run(t, []etest{ - {src: src0 + "f(1)", res: "2"}, - {src: src0 + "f(2)", res: "3"}, - {src: src0 + "f(3)", res: "5"}, - {src: src0 + "f(4)", res: "10"}, - {src: src0 + "f(5)", res: "0"}, + {src: src0 + "f(1)", res: "2"}, // #00 + {src: src0 + "f(2)", res: "3"}, // #01 + {src: src0 + "f(3)", res: "5"}, // #02 + {src: src0 + "f(4)", res: "10"}, // #03 + {src: src0 + "f(5)", res: "0"}, // #04 - {src: src1 + "f(1)", res: "2"}, - {src: src1 + "f(4)", res: "5"}, - {src: src1 + "f(6)", res: "0"}, + {src: src1 + "f(1)", res: "2"}, // #05 + {src: src1 + "f(4)", res: "5"}, // #06 + {src: src1 + "f(6)", res: "0"}, // #07 }) } @@ -175,35 +182,36 @@ func TestConst(t *testing.T) { ) ` run(t, []etest{ - {src: "const a = 1+2; a", res: "3"}, - {src: "const a, b = 1, 2; a+b", res: "3"}, - {src: "const huge = 1 << 100; const four = huge >> 98; four", res: "4"}, + {src: "const a = 1+2; a", res: "3"}, // #00 + {src: "const a, b = 1, 2; a+b", res: "3"}, // #01 + {src: "const huge = 1 << 100; const four = huge >> 98; four", res: "4"}, // #02 - {src: src0 + "c", res: "2"}, + {src: src0 + "c", res: "2"}, // #03 }) } func TestArray(t *testing.T) { run(t, []etest{ - {src: "type T []int; var t T; t", res: "[]"}, - {src: "type T [3]int; var t T; t", res: "[0 0 0]"}, - {src: "type T [3]int; var t T; t[1] = 2; t", res: "[0 2 0]"}, + {src: "type T []int; var t T; t", res: "[]"}, // #00 + {src: "type T [3]int; var t T; t", res: "[0 0 0]"}, // #01 + {src: "type T [3]int; var t T; t[1]", res: "0"}, // #02 + {src: "type T [3]int; var t T; t[1] = 2; t", res: "[0 2 0]"}, // #03 }) } func TestPointer(t *testing.T) { run(t, []etest{ - {src: "var a *int; a", res: "<nil>"}, - {src: "var a int; var b *int = &a; *b", res: "0"}, - {src: "var a int = 2; var b *int = &a; *b", res: "2"}, + {src: "var a *int; a", res: "<nil>"}, // #00 + {src: "var a int; var b *int = &a; *b", res: "0"}, // #01 + {src: "var a int = 2; var b *int = &a; *b", res: "2"}, // #02 }) } func TestStruct(t *testing.T) { run(t, []etest{ - {src: "type T struct {a string; b, c int}; var t T; t", res: "{ 0 0}"}, - {src: "type T struct {a int}; var t T; t.a", res: "0"}, - {src: "type T struct {a int}; var t T; t.a = 1; t.a", res: "1"}, + {src: "type T struct {a string; b, c int}; var t T; t", res: "{ 0 0}"}, // #00 + {src: "type T struct {a int}; var t T; t.a", res: "0"}, // #01 + {src: "type T struct {a int}; var t T; t.a = 1; t.a", res: "1"}, // #02 }) } @@ -214,27 +222,24 @@ func TestType(t *testing.T) { ) ` run(t, []etest{ - {src: "type t int; var a t = 1; a", res: "1"}, - {src: "type t = int; var a t = 1; a", res: "1"}, - {src: src0 + `var s S = "xx"; s`, res: "xx"}, + {src: "type t int; var a t = 1; a", res: "1"}, // #00 + {src: "type t = int; var a t = 1; a", res: "1"}, // #01 + {src: src0 + `var s S = "xx"; s`, res: "xx"}, // #02 }) } func TestVar(t *testing.T) { run(t, []etest{ - {src: "var a int; a", res: "0"}, - {src: "var a, b, c int; a", res: "0"}, - {src: "var a, b, c int; a + b", res: "0"}, - {src: "var a, b, c int; a + b + c", res: "0"}, - {src: "var a int = 2+1; a", res: "3"}, - {src: "var a, b int = 2, 5; a+b", res: "7"}, - {src: "var x = 5; x", res: "5"}, - {src: "var a = 1; func f() int { var a, b int = 3, 4; return a+b}; a+f()", res: "8"}, - {src: `var a = "hello"; a`, res: "hello"}, - {src: `var ( - a, b int = 4+1, 3 - c = 8 -); a+b+c`, res: "16"}, + {src: "var a int; a", res: "0"}, // #00 + {src: "var a, b, c int; a", res: "0"}, // #01 + {src: "var a, b, c int; a + b", res: "0"}, // #02 + {src: "var a, b, c int; a + b + c", res: "0"}, // #03 + {src: "var a int = 2+1; a", res: "3"}, // #04 + {src: "var a, b int = 2, 5; a+b", res: "7"}, // #05 + {src: "var x = 5; x", res: "5"}, // #06 + {src: "var a = 1; func f() int { var a, b int = 3, 4; return a+b}; a+f()", res: "8"}, // #07 + {src: `var a = "hello"; a`, res: "hello"}, // #08 + {src: `var ( a, b int = 4+1, 3; c = 8); a+b+c`, res: "16"}, // #09 }) } @@ -244,25 +249,25 @@ func TestImport(t *testing.T) { ) ` run(t, []etest{ - {src: "fmt.Println(4)", err: "symbol not found: fmt"}, - {src: `import "xxx"`, err: "package not found: xxx"}, - {src: `import "fmt"; fmt.Println(4)`, res: "<nil>"}, - {src: src0 + "fmt.Println(4)", res: "<nil>"}, - {src: `func main() {import "fmt"; fmt.Println("hello")}`, err: "unexpected import"}, - {src: `import m "fmt"; m.Println(4)`, res: "<nil>"}, - {src: `import . "fmt"; Println(4)`, res: "<nil>"}, + {src: "fmt.Println(4)", err: "missing symbol"}, // #00 + {src: `import "xxx"`, err: "package not found: xxx"}, // #01 + {src: `import "fmt"; fmt.Println(4)`, res: "<nil>"}, // #02 + {src: src0 + "fmt.Println(4)", res: "<nil>"}, // #03 + {src: `func main() {import "fmt"; fmt.Println("hello")}`, err: "unexpected import"}, // #04 + {src: `import m "fmt"; m.Println(4)`, res: "<nil>"}, // #05 + {src: `import . "fmt"; Println(4)`, res: "<nil>"}, // #06 }) } func TestComposite(t *testing.T) { run(t, []etest{ - {src: "type T struct{}; t := T{}; t", res: "{}"}, - {src: "t := struct{}{}; t", res: "{}"}, - {src: `type T struct {}; var t T; t = T{}; t`, res: "{}"}, - {src: `type T struct{N int; S string}; var t T; t = T{2, "foo"}; t`, res: `{2 foo}`}, - {src: `type T struct{N int; S string}; t := T{2, "foo"}; t`, res: `{2 foo}`}, - {src: `type T struct{N int; S string}; t := T{S: "foo"}; t`, res: `{0 foo}`}, - {src: `a := []int{}`, res: `[]`}, - // {src: `a := []int{1, 2, 3}`, res: `[1 2 3]`}, + {src: "type T struct{}; t := T{}; t", res: "{}"}, // #00 + {src: "t := struct{}{}; t", res: "{}"}, // #01 + {src: `type T struct {}; var t T; t = T{}; t`, res: "{}"}, // #02 + {src: `type T struct{N int; S string}; var t T; t = T{2, "foo"}; t`, res: `{2 foo}`}, // #03 + {src: `type T struct{N int; S string}; t := T{2, "foo"}; t`, res: `{2 foo}`}, // #04 + // {src: `type T struct{N int; S string}; t := T{S: "foo"}; t`, res: `{0 foo}`}, // #05 + // {src: `a := []int{}`, res: `[]`}, // #06 + // {src: `a := []int{1, 2, 3}`, res: `[1 2 3]`}, // #07 }) } |
