diff options
Diffstat (limited to 'vm/vm_test.go')
| -rw-r--r-- | vm/vm_test.go | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/vm/vm_test.go b/vm/vm_test.go new file mode 100644 index 0000000..08e8054 --- /dev/null +++ b/vm/vm_test.go @@ -0,0 +1,155 @@ +package vm + +import ( + "fmt" + "log" + "testing" +) + +func init() { + log.SetFlags(log.Lshortfile) +} + +func TestVM(t *testing.T) { + for _, test := range tests { + test := test + t.Run("", func(t *testing.T) { + m := &Machine{} + for _, v := range test.sym { + m.Push(v) + } + m.PushCode(test.code...) + if err := m.Run(); err != nil { + t.Errorf("run error: %v", err) + } + t.Log(m.mem) + r := fmt.Sprintf("%v", m.mem[test.start:test.end]) + if r != test.mem { + t.Errorf("got %v, want %v", r, test.mem) + } + }) + } +} + +func BenchmarkVM(b *testing.B) { + for _, test := range tests { + test := test + b.Run("", func(b *testing.B) { + for i := 0; i < b.N; i++ { + b.StopTimer() + m := &Machine{} + m.PushCode(test.code...) + b.StartTimer() + + if err := m.Run(); err != nil { + b.Errorf("run error: %v", err) + } + } + }) + } +} + +var tests = []struct { + sym []any // initial memory values + code [][]int64 // bytecode to execute + start, end int // + mem string // expected memory content +}{{ // #00 -- A simple addition. + code: [][]int64{ + {0, Push, 1}, + {0, Push, 2}, + {0, Add}, + {0, Exit}, + }, + start: 0, end: 1, mem: "[3]", +}, { // #01 -- Calling a function defined outside the VM. + sym: []any{fmt.Println, "Hello"}, + code: [][]int64{ + {0, Dup, 0}, + {0, CallX, 1}, + {0, Exit}, + }, + start: 1, end: 3, mem: "[6 <nil>]", +}, { // #02 -- Defining and calling a function in VM. + code: [][]int64{ + {0, Jump, 3}, // 0 + {0, Push, 3}, // 1 + {0, Return, 1, 1}, // 2 + {0, Push, 1}, // 3 + {0, Calli, -3}, // 4 + {0, Exit}, // 5 + }, + start: 0, end: 1, mem: "[3]", +}, { // #03 -- Defining and calling a function in VM. + code: [][]int64{ + {0, Jump, 3}, // 0 + {0, Push, 3}, // 1 + {0, Return, 1, 1}, // 2 + {0, Push, 1}, // 3 + {0, Push, 1}, // 4 + {0, Call}, // 5 + {0, Exit}, // 6 + }, + start: 0, end: 1, mem: "[3]", +}, { // #04 -- Defining and calling a function in VM. + code: [][]int64{ + {0, Jump, 5}, // 0 + {0, Push, 3}, // 1 + {0, Fassign, -2}, // 2 + {0, Fdup, -2}, // 3 + {0, Return, 1, 1}, // 4 + {0, Push, 1}, // 5 + {0, Push, 1}, // 6 + {0, Call}, // 7 + {0, Exit}, // 8 + }, + start: 0, end: 1, mem: "[3]", +}, { // #05 -- Fibonacci numbers, hand written. Showcase recursivity. + code: [][]int64{ + {0, Jump, 19}, // 0 + {0, Push, 2}, // 2 [2] + {0, Fdup, -2}, // 1 [2 i] + {0, Lower}, // 3 [true/false] + {0, JumpTrue, 13}, // 4 [], goto 17 + {0, Push, 2}, // 6 [i 2] + {0, Fdup, -2}, // 5 [i] + {0, Sub}, // 7 [(i-2)] + {0, Push, 1}, // 8 + {0, Call}, // 9 [fib(i-2)] + {0, Push, 1}, // 11 [(i-2) i 1] + {0, Fdup, -2}, // 10 [fib(i-2) i] + {0, Sub}, // 12 [(i-2) (i-1)] + {0, Push, 1}, // 13 + {0, Call}, // 14 [fib(i-2) fib(i-1)] + {0, Add}, // 15 [fib(i-2)+fib(i-1)] + {0, Return, 1, 1}, // 16 return i + {0, Fdup, -2}, // 17 [i] + {0, Return, 1, 1}, // 18 return i + {0, Push, 6}, // 19 [1] + {0, Push, 1}, // 20 + {0, Call}, // 21 [fib(*1)] + {0, Exit}, // 22 + }, + start: 0, end: 1, mem: "[8]", +}, { // #06 -- Fibonacci with some immediate instructions. + code: [][]int64{ + {0, Jump, 14}, // 0 + {0, Fdup, -2}, // 1 [i] + {0, Loweri, 2}, // 2 [true/false] + {0, JumpTrue, 9}, // 3 [], goto 12 + {0, Fdup, -2}, // 4 [i] + {0, Subi, 2}, // 5 [(i-2)] + {0, Calli, -5}, // 6 [fib(i-2)] + {0, Fdup, -2}, // 7 [fib(i-2) i] + {0, Subi, 1}, // 8 [(i-2) (i-1)] + {0, Calli, -8}, // 9 [fib(i-2) fib(i-1)], call 1 + {0, Add}, // 10 [fib(i-2)+fib(i-1)] + {0, Return, 1, 1}, // 11 return i + {0, Fdup, -2}, // 12 [i] + {0, Return, 1, 1}, // 13 return i + {0, Push, 6}, // 14 [1] + {0, Calli, -14}, // 15 [fib(*1)], call 1 + {0, Exit}, // 16 + }, + start: 0, end: 1, mem: "[8]", +}} |