diff options
Diffstat (limited to 'interp')
| -rw-r--r-- | interp/interpreter_test.go | 173 |
1 files changed, 89 insertions, 84 deletions
diff --git a/interp/interpreter_test.go b/interp/interpreter_test.go index bce97b9..e1e7107 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 { @@ -69,18 +70,24 @@ func TestExpr(t *testing.T) { }) } +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", res: "true"}, // #00 - {src: "false", res: "false"}, // #01 - {src: "true && false", res: "false"}, // #02 - {src: "true && true", res: "true"}, // #03 - {src: "true && true && false", res: "false"}, // #04 - {src: "false || true && true", res: "true"}, // #05 - {src: "2 < 3 && 1 > 2 || 3 == 3", res: "true"}, // #06 - {src: "2 > 3 && 1 > 2 || 3 == 3", res: "true"}, // #07 - {src: "2 > 3 || 2 == 1+1 && 3>0", res: "true"}, // #08 - {src: "2 > 3 || 2 == 1+1 && 3>4 || 1<2", res: "true"}, // #09 + {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 {src: "a := 1+1 < 3 && 4 == 2+2; a", res: "true"}, // #10 {src: "a := 1+1 < 3 || 3 == 2+2; a", res: "true"}, // #11 }) @@ -88,37 +95,37 @@ func TestLogical(t *testing.T) { 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 }) } @@ -157,15 +164,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 }) } @@ -177,35 +184,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 }) } @@ -216,27 +224,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 }) } @@ -246,25 +251,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 }) } |
