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 '"
