35 lines
891 B
Python
35 lines
891 B
Python
# Eryn Wells <eryn@erynwells.me>
|
|
|
|
from erynrl.geometry import Point
|
|
|
|
|
|
def test_point_neighbors():
|
|
'''Check that Point.neighbors returns all neighbors'''
|
|
test_point = Point(5, 5)
|
|
|
|
expected_neighbors = set([
|
|
Point(4, 4),
|
|
Point(5, 4),
|
|
Point(6, 4),
|
|
Point(4, 5),
|
|
# Point(5, 5),
|
|
Point(6, 5),
|
|
Point(4, 6),
|
|
Point(5, 6),
|
|
Point(6, 6),
|
|
])
|
|
|
|
neighbors = set(test_point.neighbors)
|
|
for pt in expected_neighbors:
|
|
assert pt in neighbors
|
|
|
|
assert expected_neighbors - neighbors == set(), \
|
|
f"Found some points that didn't belong in the set of neighbors of {test_point}"
|
|
|
|
|
|
def test_point_manhattan_distance():
|
|
'''Check that the Manhattan Distance calculation on Points is correct'''
|
|
point_a = Point(3, 2)
|
|
point_b = Point(8, 5)
|
|
|
|
assert point_a.manhattan_distance_to(point_b) == 8
|