class TestRace(unittest.TestCase): def setUp(self): self.test_race = Race('Dwarf') self.expected_effect = Effect() self.expected_description = 'Dwarves are a hearty breed, and make excellent warriors due to their high strength and toughness.' def tearDown(self): self.test_race = None self.expected_effect = None self.expected_description = None def test_total_hp(self): self.assertEqual(0, self.test_race.total_hp()) def test_total_strength(self): self.assertEqual(5, self.test_race.total_strength()) def test_total_intelligence(self): self.assertEqual(0, self.test_race.total_intelligence()) def test_total_agility(self): self.assertEqual(0, self.test_race.total_agility()) def test_total_attack(self): self.assertEqual(Damage(0,0), self.test_race.total_attack()) def test_total_defense(self): self.assertEqual(Damage(20,30), self.test_race.total_defense()) def test_valid_name(self): self.assertFalse(self.test_race.valid_name('INVALID')) def test_valid_name2(self): self.assertTrue(self.test_race.valid_name('DWARF')) def test_valid_name3(self): self.assertTrue(self.test_race.valid_name('dwarf')) def test_race_effect(self): self.assertEqual(self.expected_effect, self.test_race.race_effect("Dwarf")) def test_race_description(self): self.assertEqual(self.expected_description, self.test_race.race_description("Dwarf")) def test_race_description2(self): self.assertEqual(self.expected_description, self.test_race.race_description("dwarf")) def test_race_description3(self): self.assertEqual(self.expected_description, self.test_race.race_description("DWARF"))