def test_move_over_past(self):
r""" show you cant pass over a previously occupied space based on board.map """
b1 = board.Board(self.small_plain)
start = np.array((3, 3), dtype='int')
k1 = knight.Knight(b1,start)
# set move choice 0
move_choice = 0
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((5, 4), dtype='int')).all())
# try to go back in strict mode (fail)
move_choice = 3
k1.execute_move(move_choice, strict=True)
# confirm lack of movement
self.assertTrue((k1.position == np.array((5, 4), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . S . . . .
. . . x . . . .
. . . x K . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
# try to go back without strict mode
move_choice = 3
k1.execute_move(move_choice, strict=False)
# confirm lack of movement
self.assertTrue((k1.position == np.array((4, 2), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . S . . . .
. . K x . . . .
. . x x S . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
def test_valid_sequence(self):
r""" tests a valid sequence of moves """
b1 = board.Board(self.small_plain)
start = np.array((3, 3), dtype='int')
k1 = knight.Knight(b1,start)
# set move sequence
move_seq = [0, 5, 6, 3, 2]
# check sequence validity
(cost, valid, endloc) = k1.validate_sequence(move_seq)
self.assertTrue(valid)
self.assertEqual(cost, len(move_seq))
self.assertTrue((k1.position == start).all())
# change the board layout to reflect the move
for each in move_seq:
k1.execute_move(each)
self.assertTrue((k1.position == np.array((2, 3), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . x x S . .
. . . K . x x S
. . . S x S x x
. . . x x . . .
. . . x S . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
def test_good_knight_init(self):
r""" initialize knight in valid location/terrain """
b1 = board.Board(self.small_plain)
cent = np.array((3, 3), dtype='int')
k1 = knight.Knight(b1,cent)
self.assertTrue((k1.position == cent).all())
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . K . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
def test_valid_moves(self):
r""" test valid moves in order {0, 5, 6, 3, 2, 7, 1, 4} """
b1 = board.Board(self.small_plain)
start = np.array((3, 3), dtype='int')
k1 = knight.Knight(b1,start)
# set move choice 0
move_choice = 0
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((5, 4), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . S . . . .
. . . x . . . .
. . . x K . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
#
# set move choice 5
move_choice = 5
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((3, 5), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . S x K . .
. . . x x . . .
. . . x S . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
#
# set move choice 6
move_choice = 6
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((2, 7), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . K
. . . S x S x x
. . . x x . . .
. . . x S . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
#
# set move choice 3
move_choice = 3
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((1, 5), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . K . .
. . . . . x x S
. . . S x S x x
. . . x x . . .
. . . x S . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
#
# set move choice 2
move_choice = 2
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((2, 3), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . x x S . .
. . . K . x x S
. . . S x S x x
. . . x x . . .
. . . x S . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
#
# reset board
b1 = board.Board(self.small_plain)
start = np.array((3, 3), dtype='int')
k1 = knight.Knight(b1,start)
# set move choice 7
move_choice = 7
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((4, 5), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . S x x . .
. . . . . K . .
. . . . . . . .
. . . . . . . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
#
# set move choice 1
move_choice = 1
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((6, 4), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . S x x . .
. . . . . S . .
. . . . . x . .
. . . . K x . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)
#
# set move choice 4
move_choice = 4
# change the board layout to reflect the move
k1.execute_move(move_choice)
self.assertTrue((k1.position == np.array((4, 3), dtype='int')).all())
# confirm state of board
with capture_output() as (out, _):
b1.display()
my_out = out.getvalue().strip()
out.close()
out_list = [ each.strip() for each in
""". . . . . . . .
. . . . . . . .
. . . . . . . .
. . . S x x . .
. . . K x S . .
. . . . x x . .
. . . . S x . .
. . . . . . . .""".split('\n')]
expected_out = '\n'.join(out_list)
self.assertEqual(my_out, expected_out)