def test_color_rgb(): c = color.ColorRGB(128, 0, 34) assert 128 == c.r assert 0 == c.g assert 34 == c.b with pytest.raises(ValueError): color.ColorRGB(-1, 0, 34) with pytest.raises(ValueError): color.ColorRGB(0, -1, 34) with pytest.raises(ValueError): color.ColorRGB(0, 28, -1)
def test_color_rgb(self): c = color.ColorRGB(128, 0, 34) self.assertEqual(128, c.r) self.assertEqual(0, c.g) self.assertEqual(34, c.b) self.assertRaises(ValueError, color.ColorRGB, -1, 0, 34) self.assertRaises(ValueError, color.ColorRGB, 0, -1, 34) self.assertRaises(ValueError, color.ColorRGB, 0, 28, -1)
def test_colorize(): s = color.colorize('hello', color.RED, palette='ANSI') assert '\033' in s assert '[3' in s assert '1m' in s with pytest.raises(ValueError): color.colorize('hello', color.RED, palette='FOO') with pytest.raises(ValueError): color.colorize('hello', color.ColorRGB(128, 0, 34), palette='ANSI')
def test_colorize(self): s = color.colorize('hello', color.RED, palette='ANSI') self.assertIn('\033', s) self.assertIn('[3', s) self.assertIn('1m', s) with self.assertRaises(ValueError): color.colorize('hello', color.RED, palette='FOO') with self.assertRaises(ValueError): color.colorize('hello', color.ColorRGB(128, 0, 34), palette='ANSI')