def test_cycle():
m = Map(6, 6)
s = Snake(m, Direc.RIGHT, [Pos(1, 2), Pos(1, 1)],
[PointType.HEAD_R, PointType.BODY_HOR])
solver = HamiltonSolver(s, False)
table = solver.table
cnt = 0
ori_head = s.head()
while True:
head = s.head()
assert cnt == table[head.x][head.y].idx
s.move(solver.next_direc())
cnt += 1
if s.head() == ori_head:
break
assert cnt == m.capacity
def test_cycle():
m = Map(6, 6)
s = Snake(m, Direc.RIGHT,
[Pos(1, 2), Pos(1, 1)],
[PointType.HEAD_R, PointType.BODY_HOR])
solver = HamiltonSolver(s, False)
table = solver.table
cnt = 0
ori_head = s.head()
while True:
head = s.head()
assert cnt == table[head.x][head.y].idx
s.move(solver.next_direc())
cnt += 1
if s.head() == ori_head:
break
assert cnt == m.capacity
def test_dead():
m = Map(5, 5)
s = Snake(m, Direc.RIGHT,
[Pos(1, 3), Pos(1, 2), Pos(1, 1)],
[PointType.HEAD_R, PointType.BODY_HOR, PointType.BODY_HOR])
assert not s.dead
s.move(s.direc)
assert s.dead and s.len() == 3 and s.head() == Pos(1, 4)
m.reset()
s = Snake(m, Direc.RIGHT,
[Pos(1, 3), Pos(1, 2), Pos(1, 1)],
[PointType.HEAD_R, PointType.BODY_HOR, PointType.BODY_HOR])
assert not s.dead
s.move(Direc.UP)
assert s.dead and s.len() == 3 and s.head() == Pos(0, 3)
m.reset()
s = Snake(m, Direc.DOWN,
[Pos(3, 1), Pos(2, 1), Pos(1, 1)],
[PointType.HEAD_D, PointType.BODY_VER, PointType.BODY_VER])
assert not s.dead
s.move(s.direc)
assert s.dead and s.len() == 3 and s.head() == Pos(4, 1)
m.reset()
s = Snake(m, Direc.DOWN,
[Pos(3, 1), Pos(2, 1), Pos(1, 1)],
[PointType.HEAD_D, PointType.BODY_VER, PointType.BODY_VER])
assert not s.dead
s.move(Direc.LEFT)
assert s.dead and s.len() == 3 and s.head() == Pos(3, 0)
m.reset()
s = Snake(
m, Direc.LEFT,
[Pos(2, 2), Pos(3, 2),
Pos(3, 1), Pos(2, 1),
Pos(1, 1)], [
PointType.HEAD_U, PointType.BODY_LU, PointType.BODY_UR,
PointType.BODY_VER, PointType.BODY_VER
])
assert not s.dead
s.move(s.direc)
assert s.dead and s.len() == 5 and s.head() == Pos(2, 1)
def test_init():
m = Map(5, 5)
s = Snake(m, Direc.RIGHT,
[Pos(1, 3), Pos(1, 2), Pos(1, 1)],
[PointType.HEAD_R, PointType.BODY_HOR, PointType.BODY_HOR])
assert not s.dead
assert s.direc is Direc.RIGHT
assert s.len() == 3
assert s.head() == Pos(1, 3)
assert s.bodies[1] == Pos(1, 2)
assert s.tail() == Pos(1, 1)
assert m.point(Pos(1, 1)).type == PointType.BODY_HOR
assert m.point(Pos(1, 2)).type == PointType.BODY_HOR
assert m.point(Pos(1, 3)).type == PointType.HEAD_R
def test_move_eat():
m = Map(5, 5)
s = Snake(m, Direc.RIGHT, [Pos(1, 2), Pos(1, 1)],
[PointType.HEAD_R, PointType.BODY_HOR])
assert s.len() == 2
m.createFood(Pos(1, 3))
assert m.hasFood()
s.move(Direc.RIGHT)
assert not m.hasFood()
assert s.head() == Pos(1, 3) and s.tail() == Pos(1, 1)
assert m.point(s.tail()).type == PointType.BODY_HOR
assert m.point(Pos(1, 2)).type == PointType.BODY_HOR
assert m.point(s.head()).type == PointType.HEAD_R
s.move(Direc.DOWN)
assert s.head() == Pos(2, 3) and s.tail() == Pos(1, 2)
assert m.point(s.tail()).type == PointType.BODY_HOR
assert m.point(Pos(1, 3)).type == PointType.BODY_DL
assert m.point(s.head()).type == PointType.HEAD_D
s.move(Direc.LEFT)
assert s.head() == Pos(2, 2) and s.tail() == Pos(1, 3)
assert m.point(s.tail()).type == PointType.BODY_DL
assert m.point(Pos(2, 3)).type == PointType.BODY_LU
assert m.point(s.head()).type == PointType.HEAD_L
s.move(Direc.LEFT)
assert s.head() == Pos(2, 1) and s.tail() == Pos(2, 3)
assert m.point(s.tail()).type == PointType.BODY_LU
assert m.point(Pos(2, 2)).type == PointType.BODY_HOR
assert m.point(s.head()).type == PointType.HEAD_L
s.move(Direc.DOWN)
assert s.head() == Pos(3, 1) and s.tail() == Pos(2, 2)
assert m.point(s.tail()).type == PointType.BODY_HOR
assert m.point(Pos(2, 1)).type == PointType.BODY_RD
assert m.point(s.head()).type == PointType.HEAD_D
s.move(Direc.RIGHT)
assert s.head() == Pos(3, 2) and s.tail() == Pos(2, 1)
assert m.point(s.tail()).type == PointType.BODY_RD
assert m.point(Pos(3, 1)).type == PointType.BODY_UR
assert m.point(s.head()).type == PointType.HEAD_R
s.move(Direc.RIGHT)
assert s.head() == Pos(3, 3) and s.tail() == Pos(3, 1)
assert m.point(s.tail()).type == PointType.BODY_UR
assert m.point(Pos(3, 2)).type == PointType.BODY_HOR
assert m.point(s.head()).type == PointType.HEAD_R
s.move(Direc.UP)
assert s.head() == Pos(2, 3) and s.tail() == Pos(3, 2)
assert m.point(s.tail()).type == PointType.BODY_HOR
assert m.point(Pos(3, 3)).type == PointType.BODY_LU
assert m.point(s.head()).type == PointType.HEAD_U
s.move(Direc.LEFT)
assert s.head() == Pos(2, 2) and s.tail() == Pos(3, 3)
assert m.point(s.tail()).type == PointType.BODY_LU
assert m.point(Pos(2, 3)).type == PointType.BODY_DL
assert m.point(s.head()).type == PointType.HEAD_L
s.move(Direc.LEFT)
assert s.head() == Pos(2, 1) and s.tail() == Pos(2, 3)
assert m.point(s.tail()).type == PointType.BODY_DL
assert m.point(Pos(2, 2)).type == PointType.BODY_HOR
assert m.point(s.head()).type == PointType.HEAD_L
s.move(Direc.UP)
assert s.head() == Pos(1, 1) and s.tail() == Pos(2, 2)
assert m.point(s.tail()).type == PointType.BODY_HOR
assert m.point(Pos(2, 1)).type == PointType.BODY_UR
assert m.point(s.head()).type == PointType.HEAD_U
s.move(Direc.RIGHT)
assert s.head() == Pos(1, 2) and s.tail() == Pos(2, 1)
assert m.point(s.tail()).type == PointType.BODY_UR
assert m.point(Pos(1, 1)).type == PointType.BODY_RD
assert m.point(s.head()).type == PointType.HEAD_R
s.move(Direc.RIGHT)
assert s.head() == Pos(1, 3) and s.tail() == Pos(1, 1)
assert m.point(s.tail()).type == PointType.BODY_RD
assert m.point(Pos(1, 2)).type == PointType.BODY_HOR
assert m.point(s.head()).type == PointType.HEAD_R
s.move(Direc.DOWN)
s.move(Direc.DOWN)
assert s.head() == Pos(3, 3) and s.tail() == Pos(1, 3)
assert m.point(s.tail()).type == PointType.BODY_DL
assert m.point(Pos(2, 3)).type == PointType.BODY_VER
assert m.point(s.head()).type == PointType.HEAD_D
s.move(Direc.LEFT)
s.move(Direc.LEFT)
s.move(Direc.UP)
s.move(Direc.UP)
assert s.head() == Pos(1, 1) and s.tail() == Pos(3, 1)
assert m.point(s.tail()).type == PointType.BODY_UR
assert m.point(Pos(2, 1)).type == PointType.BODY_VER
assert m.point(s.head()).type == PointType.HEAD_U
assert s.len() == 3
# Eat full
assert not m.isFull()
food_pos = [
Pos(1, 2),
Pos(2, 2),
Pos(3, 2),
Pos(3, 3),
Pos(2, 3),
Pos(1, 3)
]
move_direc = [
Direc.RIGHT, Direc.DOWN, Direc.DOWN, Direc.RIGHT, Direc.UP, Direc.UP
]
for i, pos in enumerate(food_pos):
m.createFood(pos)
s.move(move_direc[i])
assert m.isFull()
assert s.len() == 9 and s.steps == 25 and not s.dead
