class PieceTestCase(unittest.TestCase):
def setUp(self):
self.white_piece = Piece(Color.white, (0,0), None)
self.black_piece = Piece(Color.black, (0,0), None)
def test_piece_as_str(self):
self.assertEqual("B", str(self.black_piece))
self.assertEqual("W", str(self.white_piece))
def test_piece_get_neighbors(self):
expected = sorted([(-1, 0), (1, 0), (0, -1), (0, 1)])
result = sorted(self.white_piece.get_neighbors())
self.assertListEqual(expected, result)
def test_is_ally(self):
self.assertFalse(self.white_piece.is_ally(self.black_piece))
def test_is_enemy(self):
self.assertTrue(self.white_piece.is_enemy(self.black_piece))
