From 126a628b171e2a7d2490290185ab21990b0d9698 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Mon, 12 Dec 2022 10:49:04 +0100 Subject: Day12 [Wip] --- day12/test_init.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 day12/test_init.py (limited to 'day12/test_init.py') diff --git a/day12/test_init.py b/day12/test_init.py new file mode 100644 index 0000000..f721d56 --- /dev/null +++ b/day12/test_init.py @@ -0,0 +1,27 @@ +# -*- coding: utf-8 -*- +from day12 import Map + + +class TestMap: + def test_neighbours(self): + + map_input = ["".join([str(i) for i in range(8)]) for _ in range(5)] + + map = Map(map=map_input) + + assert set(map.neighbours(0, 0)) == { + (0, 1), + (1, 0), + } + + assert set(map.neighbours(0, 4)) == { + (0, 3), + (1, 4), + } + + assert set(map.neighbours(1, 1)) == { + (0, 1), + (1, 0), + (1, 2), + (2, 1), + } -- cgit v1.2.3