class getHeader(unittest.TestCase): """ Test cases of getFirstPokemon """ def setUp(self): """ Build the Trainer and Pokemon lists for use in tests """ self.trainer = ComputerTrainer() def headerIsCompHeader(self): """ Check that the attack returned by getAction is an attack the Pokemon has """ assert self.trainer.getHeader() is ComputerTrainer.header, "Should be 'Enemy '"