1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
# -*- coding: utf-8 -*-
from aoc.datastructures import Coordinate, Coordinate3
class TestCoordinate:
def test_addition(self):
c1 = Coordinate(1, 1)
c2 = Coordinate(2, 2)
assert c1 + c2 == Coordinate(3, 3)
def test_subtraction(self):
c1 = Coordinate(2, 2)
c2 = Coordinate(1, 1)
assert c1 - c2 == Coordinate(1, 1)
def test_manhattan_distance(self):
c1 = Coordinate(1, 1)
c2 = Coordinate(3, 3)
assert c1.manhattan_distance(c2) == 4
def test_polarity(self):
c1 = Coordinate(2, -3)
assert c1.polarity == Coordinate(1, -1)
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),
}
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),
}
assert set(c.neighbours(no_diagonal=True)) == neighbours_no_diagonal
class TestCoordinate3:
def test_addition(self):
c1 = Coordinate3(1, 1, 1)
c2 = Coordinate3(2, 2, 2)
assert c1 + c2 == Coordinate3(3, 3, 3)
def test_subtraction(self):
c1 = Coordinate3(2, 2, 2)
c2 = Coordinate3(1, 1, 1)
assert c1 - c2 == Coordinate3(1, 1, 1)
def test_manhattan_distance(self):
c1 = Coordinate3(1, 1, 1)
c2 = Coordinate3(3, 3, 3)
assert c1.manhattan_distance(c2) == 6
def test_polarity(self):
c1 = Coordinate3(2, -3, 0)
assert c1.polarity == Coordinate3(1, -1, 0)
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),
}
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),
}
assert set(c.neighbours(no_diagonal=True)) == neighbours_no_diagonal
|
