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'))
def test_red_with_colored_on(self):
display = Display()
display.colored = True
self.assertEqual(display.red('foobar'), colored('foobar', 'red'))
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'))
def test_cyan_with_colored_on(self):
display = Display()
display.colored = True
self.assertEqual(display.cyan('foobar'), colored('foobar', 'cyan'))