class test_player_masking(unittest.TestCase): """Check that the game correctly masks the current player""" def setUp(self): self.controller = Connect_Four_Controller() print("Running: ", str(self._testMethodName) + "\n " + str(self.shortDescription()) + "\n") def tearDown(self): del self.controller def test_mask_player(self): """Test if the players integer values equal their color""" #set the player int to the 'Black' designation self.controller.game_state.player = -1 #test the player int is equal to the player color self.assertEqual(self.controller.mask_player(self.controller.game_state.get_current_player()), "Black")