aoc-2019/intcode/vm_test.go

package intcode

import (
	"reflect"
	"sync"
	"testing"
)

func TestDayTwoSamples(t *testing.T) {
	tables := []struct {
		given    []int
		expected []int
	}{
		{[]int{1, 0, 0, 0, 99}, []int{2, 0, 0, 0, 99}},
		{[]int{2, 3, 0, 3, 99}, []int{2, 3, 0, 6, 99}},
		{[]int{2, 4, 4, 5, 99, 0}, []int{2, 4, 4, 5, 99, 9801}},
		{[]int{1, 1, 1, 4, 99, 5, 6, 0, 99}, []int{30, 1, 1, 4, 2, 5, 6, 0, 99}},
	}

	for _, table := range tables {
		vm := NewVirtualMachine(table.given)
		vm.Run()
		if !reflect.DeepEqual(table.expected, vm.Memory) {
			t.Errorf("Evaluation of %v was incorrect, got: %v, want: %v.", table.given, vm.Memory, table.expected)
		}
	}
}

func TestDayFiveSamples(t *testing.T) {
	tables := []struct {
		given  []int
		input  []int
		output []int
	}{
		// Reads then outputs a number
		{[]int{3, 0, 4, 0, 99}, []int{123}, []int{123}},
		// Using position mode, consider whether the input is equal to 8; output 1 (if it is) or 0 (if it is not).
		{[]int{3, 9, 8, 9, 10, 9, 4, 9, 99, -1, 8}, []int{8}, []int{1}},
		{[]int{3, 9, 8, 9, 10, 9, 4, 9, 99, -1, 8}, []int{7}, []int{0}},
		// Using position mode, consider whether the input is less than 8; output 1 (if it is) or 0 (if it is not).
		{[]int{3, 9, 7, 9, 10, 9, 4, 9, 99, -1, 8}, []int{7}, []int{1}},
		{[]int{3, 9, 7, 9, 10, 9, 4, 9, 99, -1, 8}, []int{8}, []int{0}},
		// Using immediate mode, consider whether the input is equal to 8; output 1 (if it is) or 0 (if it is not).
		{[]int{3, 3, 1108, -1, 8, 3, 4, 3, 99}, []int{8}, []int{1}},
		{[]int{3, 3, 1108, -1, 8, 3, 4, 3, 99}, []int{9}, []int{0}},
		// Using immediate mode, consider whether the input is less than 8; output 1 (if it is) or 0 (if it is not).
		{[]int{3, 3, 1107, -1, 8, 3, 4, 3, 99}, []int{0}, []int{1}},
		{[]int{3, 3, 1107, -1, 8, 3, 4, 3, 99}, []int{10}, []int{0}},
	}

	for n, table := range tables {
		vm := NewVirtualMachine(table.given)
		vm.Input = make(chan int, 1)
		vm.Output = make(chan int, 1)

		for _, v := range table.input {
			vm.Input <- v
		}

		vm.Run()

		for _, v := range table.output {
			actual := <-vm.Output
			if !reflect.DeepEqual(v, actual) {
				t.Errorf("[%d] Wrong output value received for %v, got: %v, want: %v.", n, table.given, actual, v)
			}
		}
	}
}

func TestDayNineSamples(t *testing.T) {
	tables := []struct {
		given  []int
		input  []int
		output []int
	}{
		// Takes no input and produces a copy of itself as output.
		{[]int{109, 1, 204, -1, 1001, 100, 1, 100, 1008, 100, 16, 101, 1006, 101, 0, 99}, []int{}, []int{109, 1, 204, -1, 1001, 100, 1, 100, 1008, 100, 16, 101, 1006, 101, 0, 99}},
		// Should output a 16 digit number.
		{[]int{1102, 34915192, 34915192, 7, 4, 7, 99, 0}, []int{}, []int{1219070632396864}},
		// Should output the large number in the middle.
		{[]int{104, 1125899906842624, 99}, []int{}, []int{1125899906842624}},
	}

	for _, table := range tables {
		vm := NewVirtualMachine(table.given)
		vm.Input = make(chan int, 1)
		vm.Output = make(chan int, 100)
		wg := &sync.WaitGroup{}
		wg.Add(1)

		go func() {
			vm.Run()
			wg.Done()
		}()

		for _, v := range table.input {
			vm.Input <- v
		}

		wg.Wait()

		for _, v := range table.output {
			actual := <-vm.Output
			if !reflect.DeepEqual(v, actual) {
				t.Errorf("Wrong output value received for %v, got: %v, want: %v.", table.given, actual, v)
			}
		}
	}
}