1 files changed, 16 insertions, 0 deletions

diff --git a/aoc/tests/test_datastructures.py b/aoc/tests/test_datastructures.py
new file mode 100644
index 0000000..a7965f5
--- /dev/null
+++ b/aoc/tests/test_datastructures.py
@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+from aoc.datastructures import Coordinate3
+
+
+class TestCoordinate3:
+    def test_neighbours_nodiagonal(self):
+        coordinate = Coordinate3(0, 0, 0)
+
+        assert set(coordinate.neighbours(True)) == {
+            Coordinate3(-1, 0, 0),
+            Coordinate3(1, 0, 0),
+            Coordinate3(0, -1, 0),
+            Coordinate3(0, 1, 0),
+            Coordinate3(0, 0, -1),
+            Coordinate3(0, 0, 1),
+        }