diff --git a/tests/test_colors.py b/tests/test_colors.py index 235b0e24..fe9b873a 100644 --- a/tests/test_colors.py +++ b/tests/test_colors.py @@ -9,12 +9,14 @@ class TestColors(unittest.TestCase): self.color = colors() def test_colors(self): - self.assertIn('BOLD', self.color) - self.assertIn('RED', self.color) - self.assertIn('YELLOW', self.color) - self.assertIn('GREEN', self.color) - self.assertIn('BLUE', self.color) - self.assertIn('GREY', self.color) + self.assertIn('bold', self.color) + self.assertIn('red', self.color) + self.assertIn('yellow', self.color) + self.assertIn('cyan', self.color) + self.assertIn('green', self.color) + self.assertIn('blue', self.color) + self.assertIn('grey', self.color) + self.assertIn('endc', self.color) if __name__ == '__main__':