class TestSpecies(unittest.TestCase): def setUp(self): self.attacker = Species() self.attacker.traits = [TraitCard(CARNIVORE)] self.defender = Species() self.left_neighbor = Species() self.right_neighbor = Species() self.species_1 = Species(4, 4, 4) self.species_2 = Species(4, 4, 4) self.species_3 = Species(4, 4, 3) self.species_4 = Species(4, 3, 3) self.species_5 = Species(3, 3, 3) self.species_list = [ self.species_2, self.species_4, self.species_3, self.species_5, self.species_1 ] self.player_1 = PlayerState(1) self.player_2 = PlayerState(2) def test_all_attackable_species(self): self.species_1.traits = [TraitCard(CARNIVORE)] self.species_2.traits = [TraitCard(CLIMBING)] self.species_4.traits = [TraitCard(HARDSHELL)] self.player_1.species = [self.species_2, self.species_3] self.player_2.species = [self.species_4, self.species_5] self.assertEqual( self.species_1.all_attackable_species( [self.player_1, self.player_2]), [self.species_3, self.species_5]) def test_attackable_species(self): self.species_1.traits = [TraitCard(CARNIVORE)] self.species_2.traits = [TraitCard(CLIMBING)] self.species_4.traits = [TraitCard(HARDSHELL)] self.player_1.species = [ self.species_2, self.species_3, self.species_4, self.species_5 ] self.assertEqual(self.species_1.attackable_species(self.player_1), [self.species_3, self.species_5]) def test_trait_names(self): self.assertEqual(self.defender.trait_names(), []) self.assertEqual(self.attacker.trait_names(), [CARNIVORE]) def test_attackable(self): self.assertTrue(self.defender.is_attackable(self.attacker)) def test_no_carnivore(self): self.attacker.traits = [] self.assertFalse(self.defender.is_attackable(self.attacker)) def test_burrowing(self): self.defender.traits = [TraitCard(BURROWING)] self.defender.population, self.defender.food = (4, 4) self.assertFalse(self.defender.is_attackable(self.attacker)) self.defender.food = 3 self.assertTrue(self.defender.is_attackable(self.attacker)) def test_climbing(self): self.defender.traits = [TraitCard(CLIMBING)] self.assertFalse(self.defender.is_attackable(self.attacker)) self.attacker.traits.append(TraitCard(CLIMBING)) self.assertTrue(self.defender.is_attackable(self.attacker)) def test_hard_shell(self): self.defender.traits = [TraitCard(HARDSHELL)] self.defender.body = 3 self.attacker.body = 6 self.assertFalse(self.defender.is_attackable(self.attacker)) self.attacker.body = 7 self.assertTrue(self.defender.is_attackable(self.attacker)) def test_herding(self): self.defender.traits = [TraitCard(HERDING)] self.defender.population = 4 self.attacker.population = 3 self.assertFalse(self.defender.is_attackable(self.attacker)) self.attacker.population = 5 self.assertTrue(self.defender.is_attackable(self.attacker)) def test_symbiosis(self): self.defender.traits = [TraitCard(SYMBIOSIS)] self.defender.body = 3 self.right_neighbor.body = 5 self.assertFalse( self.defender.is_attackable(self.attacker, right_neighbor=self.right_neighbor)) self.right_neighbor.body = 2 self.assertTrue( self.defender.is_attackable(self.attacker, right_neighbor=self.right_neighbor)) def test_warning_call(self): self.left_neighbor.traits = [TraitCard(WARNINGCALL)] self.right_neighbor.traits = [TraitCard(WARNINGCALL)] self.assertFalse( self.defender.is_attackable(self.attacker, left_neighbor=self.left_neighbor)) self.assertFalse( self.defender.is_attackable(self.attacker, right_neighbor=self.right_neighbor)) self.attacker.traits.append(TraitCard(AMBUSH)) self.assertTrue( self.defender.is_attackable(self.attacker, left_neighbor=self.left_neighbor)) def test_show_changes(self): self.assertEquals(self.species_1.show_changes(self.species_3), '[[body, 4->3]]') self.assertEquals(self.species_1.show_changes(self.species_2), '') self.species_2.traits = [TraitCard(CLIMBING)] self.assertEquals(self.species_1.show_changes(self.species_2), '[[traits: new cards: [climbing, 0]]]') self.species_3.traits = [TraitCard(HERDING)] self.assertEquals( self.species_2.show_changes(self.species_3), '[[body, 4->3], [traits: [0, [climbing, 0]->[herding, 0]]]]') self.species_3.population = 1 self.species_3.food = 0 self.species_3.traits.append(TraitCard(BURROWING)) self.assertEquals( self.species_2.show_changes(self.species_3), '[[population, 4->1], [food, 4->0], [body, 4->3], ' '[traits: new cards: [herding, 0], [burrowing, 0]]]') self.species_2.traits = [TraitCard(CLIMBING)] self.species_3.traits = [TraitCard(CLIMBING)] self.assertEquals(self.species_2.show_changes(self.species_3), '[[population, 4->1], [food, 4->0], [body, 4->3]]')