mirror of
https://github.com/Crocmagnon/advent-of-code.git
synced 2024-11-05 14:23:58 +01:00
38 lines
905 B
Python
38 lines
905 B
Python
|
import pytest
|
||
|
|
from day14_docking import ProgramPart2
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def program():
|
||
|
|
return ProgramPart2([])
|
||
|
|
|
||
|
|
|
||
|
|
def test_no_floating(program):
|
||
|
|
program.mask = "0" * 36
|
||
|
|
assert list(program.get_masked_addresses(8)) == [8]
|
||
|
|
|
||
|
|
|
||
|
|
def test_last_floating(program):
|
||
|
|
program.mask = "0" * 35 + "X"
|
||
|
|
assert list(program.get_masked_addresses(8)) == [8, 9]
|
||
|
|
|
||
|
|
|
||
|
|
def test_second_to_last_floating(program):
|
||
|
|
program.mask = "0" * 34 + "X0"
|
||
|
|
assert list(program.get_masked_addresses(8)) == [8, 10]
|
||
|
|
|
||
|
|
|
||
|
|
def test_last_two_floating(program):
|
||
|
|
program.mask = "0" * 34 + "XX"
|
||
|
|
assert list(program.get_masked_addresses(8)) == [8, 9, 10, 11]
|
||
|
|
|
||
|
|
|
||
|
|
def test_one_bit_replacement(program):
|
||
|
|
program.mask = "0" * 35 + "1"
|
||
|
|
assert list(program.get_masked_addresses(8)) == [9]
|
||
|
|
|
||
|
|
|
||
|
|
def test_two_bits_replacement(program):
|
||
|
|
program.mask = "0" * 34 + "11"
|
||
|
|
assert list(program.get_masked_addresses(8)) == [11]
|