adventofcode/2019/tests/test_intcode.cpp

#include <gtest/gtest.h>
#include "utils.hpp"

using aoc2019::run_intcode;
using aoc2019::IntCodeComputer;

auto run_program(std::vector<int64_t> program, std::deque<int64_t> input) {
    std::deque<std::int64_t> output;
    IntCodeComputer computer(std::move(program), std::move(input));
    computer.connectOutput(output);
    computer.run();
    return output;
}

TEST(Intcode, TestReproduceInput) {
    const std::vector<int64_t> program = {109, 1, 204, -1, 1001, 100, 1, 100, 1008, 100, 16, 101, 1006, 101, 0, 99};
    const std::deque<int64_t> expected(program.begin(), program.end());

    ASSERT_EQ(expected, run_program(program, {}));
}

TEST(Intcode, TestLargeMultiplication) {
    const std::vector<int64_t> program = {1102, 34915192, 34915192, 7, 4, 7, 99, 0};

    ASSERT_EQ(1219070632396864, run_program(program, {}).front());
}

TEST(Intcode, TestLargeNumber) {
    const std::vector<int64_t> program = {104, 1125899906842624, 99};

    ASSERT_EQ(1125899906842624, run_program(program, {}).front());
}

TEST(Intcode, TestPositionEquality) {
    const std::vector<int64_t> program = {3, 9, 8, 9, 10, 9, 4, 9, 99, -1, 8};

    ASSERT_EQ(1, run_program(program, {8}).front());
    ASSERT_EQ(0, run_program(program, {9}).front());
}

TEST(Intcode, TestPositionLess) {
    const std::vector<int64_t> program = {3, 9, 7, 9, 10, 9, 4, 9, 99, -1, 8};

    ASSERT_EQ(1, run_program(program, {7}).front());
    ASSERT_EQ(0, run_program(program, {9}).front());
}

TEST(Intcode, TestImmediateEquality) {
    const std::vector<int64_t> program = {3, 3, 1108, -1, 8, 3, 4, 3, 99};

    ASSERT_EQ(1, run_program(program, {8}).front());
    ASSERT_EQ(0, run_program(program, {9}).front());
}

TEST(Intcode, TestImmediateLess) {
    const std::vector<int64_t> program = {3, 3, 1107, -1, 8, 3, 4, 3, 99};

    ASSERT_EQ(1, run_program(program, {7}).front());
    ASSERT_EQ(0, run_program(program, {9}).front());
}

TEST(Intcode, TestComplicatedConditional) {
    const std::vector<std::int64_t> program = {3, 21, 1008, 21, 8, 20, 1005, 20, 22, 107, 8, 21, 20, 1006, 20, 31,
                                               1106, 0, 36, 98, 0, 0, 1002, 21, 125, 20, 4, 20, 1105, 1, 46, 104,
                                               999, 1105, 1, 46, 1101, 1000, 1, 20, 4, 20, 1105, 1, 46, 98, 99};
    auto pcopy = program;

    auto output = run_intcode(pcopy, {7});
    ASSERT_EQ(999, output.front());

    pcopy = program;
    output = run_intcode(pcopy, {9});
    ASSERT_EQ(1001, output.front());

    pcopy = program;
    output = run_intcode(pcopy, {8});
    ASSERT_EQ(1000, output.front());
}