From e688c2b674fc7ad6a964a48df379e5abd01843a7 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Mon, 4 Dec 2023 10:40:34 +0100 Subject: Day4 --- aoc/tests/test_datastructures.py | 63 ++++++++++++++++++++++++++++++---------- 1 file changed, 48 insertions(+), 15 deletions(-) (limited to 'aoc/tests/test_datastructures.py') diff --git a/aoc/tests/test_datastructures.py b/aoc/tests/test_datastructures.py index 42cc44d..1c291cc 100644 --- a/aoc/tests/test_datastructures.py +++ b/aoc/tests/test_datastructures.py @@ -24,13 +24,26 @@ class TestCoordinate: def test_neighbours(self): c = Coordinate(0, 0) - neighbours = {Coordinate(-1, 0), Coordinate(1, 0), Coordinate(0, -1), Coordinate(0, 1), - Coordinate(-1, -1), Coordinate(-1, 1), Coordinate(1, -1), Coordinate(1, 1)} + neighbours = { + Coordinate(-1, 0), + Coordinate(1, 0), + Coordinate(0, -1), + Coordinate(0, 1), + Coordinate(-1, -1), + Coordinate(-1, 1), + Coordinate(1, -1), + Coordinate(1, 1), + } assert set(c.neighbours()) == neighbours def test_neighbours_no_diagonal(self): c = Coordinate(0, 0) - neighbours_no_diagonal = {Coordinate(-1, 0), Coordinate(1, 0), Coordinate(0, -1), Coordinate(0, 1)} + neighbours_no_diagonal = { + Coordinate(-1, 0), + Coordinate(1, 0), + Coordinate(0, -1), + Coordinate(0, 1), + } assert set(c.neighbours(no_diagonal=True)) == neighbours_no_diagonal @@ -57,23 +70,43 @@ class TestCoordinate3: def test_neighbours(self): c = Coordinate3(0, 0, 0) neighbours = { - Coordinate3(-1, -1, -1), Coordinate3(-1, -1, 0), Coordinate3(-1, -1, 1), - Coordinate3(-1, 0, -1), Coordinate3(-1, 0, 0), Coordinate3(-1, 0, 1), - Coordinate3(-1, 1, -1), Coordinate3(-1, 1, 0), Coordinate3(-1, 1, 1), - Coordinate3(0, -1, -1), Coordinate3(0, -1, 0), Coordinate3(0, -1, 1), - Coordinate3(0, 0, -1), Coordinate3(0, 0, 1), - Coordinate3(0, 1, -1), Coordinate3(0, 1, 0), Coordinate3(0, 1, 1), - Coordinate3(1, -1, -1), Coordinate3(1, -1, 0), Coordinate3(1, -1, 1), - Coordinate3(1, 0, -1), Coordinate3(1, 0, 0), Coordinate3(1, 0, 1), - Coordinate3(1, 1, -1), Coordinate3(1, 1, 0), Coordinate3(1, 1, 1) + Coordinate3(-1, -1, -1), + Coordinate3(-1, -1, 0), + Coordinate3(-1, -1, 1), + Coordinate3(-1, 0, -1), + Coordinate3(-1, 0, 0), + Coordinate3(-1, 0, 1), + Coordinate3(-1, 1, -1), + Coordinate3(-1, 1, 0), + Coordinate3(-1, 1, 1), + Coordinate3(0, -1, -1), + Coordinate3(0, -1, 0), + Coordinate3(0, -1, 1), + Coordinate3(0, 0, -1), + Coordinate3(0, 0, 1), + Coordinate3(0, 1, -1), + Coordinate3(0, 1, 0), + Coordinate3(0, 1, 1), + Coordinate3(1, -1, -1), + Coordinate3(1, -1, 0), + Coordinate3(1, -1, 1), + Coordinate3(1, 0, -1), + Coordinate3(1, 0, 0), + Coordinate3(1, 0, 1), + Coordinate3(1, 1, -1), + Coordinate3(1, 1, 0), + Coordinate3(1, 1, 1), } assert set(c.neighbours()) == neighbours def test_neighbours_no_diagonal(self): c = Coordinate3(0, 0, 0) neighbours_no_diagonal = { - Coordinate3(-1, 0, 0), Coordinate3(1, 0, 0), - Coordinate3(0, -1, 0), Coordinate3(0, 1, 0), - Coordinate3(0, 0, -1), Coordinate3(0, 0, 1) + Coordinate3(-1, 0, 0), + Coordinate3(1, 0, 0), + Coordinate3(0, -1, 0), + Coordinate3(0, 1, 0), + Coordinate3(0, 0, -1), + Coordinate3(0, 0, 1), } assert set(c.neighbours(no_diagonal=True)) == neighbours_no_diagonal -- cgit v1.2.3