class TestBoard(unittest.TestCase): def setUp(self): self.size = (100, 200) self.name = 'Test dungeon' self.board = Board(self.name, self.size) def test_create_pawn(self): pos = 3, 4 size = 5, 6 pawn = self.board.create_pawn('Dragon', 35, pos, size) self.assertIn(pawn.name, self.board.pawns) self.assertIs(pawn, self.board.pawns[pawn.name]) self.assertIn(pawn, self.board.initiatives) def del_pawn(self): pos = 3, 4 size = 5, 6 self.board.create_pawn('Dragon', 35, pos, size) pawn = self.board.del_pawn('Dragon') self.assertNotIn(pawn.name, self.board.pawns) self.assertIsNot(pawn, self.board.pawns[pawn.name]) self.assertNotIn(pawn, self.board.initiatives)
def setUp(self): self.size = (100, 200) self.name = 'Test dungeon' self.board = Board(self.name, self.size)