def test_red_without_colored_on(self): display = Display() display.colored = False self.assertEqual(display.red('foobar'), 'foobar')
def test_red_with_colored_on(self): display = Display() display.colored = True self.assertEqual(display.red('foobar'), colored('foobar', 'red'))