class TestDie(unittest.TestCase): def setUp(self): self.die_green = Die(3, 2, 1, "Green", "\033[92m") self.die_yellow = Die(2, 2, 2, "Yellow", "\033[93m") self.die_red = Die(1, 2, 3, "Red", "\033[91m") def test_die_color(self): self.assertEqual(self.die_green.color, "Green") self.assertEqual(self.die_yellow.color, "Yellow") self.assertEqual(self.die_red.color, "Red") def test_die_brains(self): self.assertEqual(self.die_green.brains, 3) self.assertEqual(self.die_yellow.brains, 2) self.assertEqual(self.die_red.brains, 1) def test_die_footprint(self): self.assertEqual(self.die_green.footprint, 2) self.assertEqual(self.die_yellow.footprint, 2) self.assertEqual(self.die_red.footprint, 2) def test_die_shotgun(self): self.assertEqual(self.die_green.shotgun, 1) self.assertEqual(self.die_yellow.shotgun, 2) self.assertEqual(self.die_red.shotgun, 3) def test_die_txt_color(self): self.assertEqual(self.die_green.txt_color, "\033[92m") self.assertEqual(self.die_yellow.txt_color, "\033[93m") self.assertEqual(self.die_red.txt_color, "\033[91m") def test_die_get_side(self): self.assertIn(self.die_green.get_side(), ["Brains", "Footprint", "Shotgun"]) self.assertIn(self.die_yellow.get_side(), ["Brains", "Footprint", "Shotgun"]) self.assertIn(self.die_red.get_side(), ["Brains", "Footprint", "Shotgun"]) def test_die_create_list(self): self.assertListEqual( self.die_green.list_of_side, ["Brains", "Brains", "Brains", "Footprint", "Footprint", "Shotgun"] ) self.assertListEqual( self.die_yellow.list_of_side, ["Brains", "Brains", "Footprint", "Footprint", "Shotgun", "Shotgun"] ) self.assertListEqual( self.die_red.list_of_side, ["Brains", "Footprint", "Footprint", "Shotgun", "Shotgun", "Shotgun"] )
def setUp(self): self.die_green = Die(3, 2, 1, "Green", "\033[92m") self.die_yellow = Die(2, 2, 2, "Yellow", "\033[93m") self.die_red = Die(1, 2, 3, "Red", "\033[91m")