def test_yellow_without_colored_on(self): display = Display() display.colored = False self.assertEqual(display.yellow('foobar'), 'foobar')
def test_yellow_with_colored_on(self): display = Display() display.colored = True self.assertEqual(display.yellow('foobar'), colored('foobar', 'yellow'))