def test_cyan_without_colored_on(self): display = Display() display.colored = False self.assertEqual(display.cyan('foobar'), 'foobar')
def test_cyan_with_colored_on(self): display = Display() display.colored = True self.assertEqual(display.cyan('foobar'), colored('foobar', 'cyan'))