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