def test_grey_without_colored_on(self): display = Display() display.colored = False self.assertEqual(display.grey('foobar'), 'foobar')
def test_grey_with_colored_on(self): display = Display() display.colored = True self.assertEqual(display.grey('foobar'), colored('foobar', 'white'))