27 lines
1,019 B
Python
27 lines
1,019 B
Python
|
|
||
|
|
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from suou import chalk
|
||
|
|
|
||
|
|
class TestColor(unittest.TestCase):
|
||
|
|
def setUp(self) -> None:
|
||
|
|
...
|
||
|
|
def tearDown(self) -> None:
|
||
|
|
...
|
||
|
|
|
||
|
|
def test_chalk_colors(self):
|
||
|
|
strg = "The quick brown fox jumps over the lazy dog"
|
||
|
|
|
||
|
|
self.assertEqual(f'\x1b[31m{strg}\x1b[39m', chalk.red(strg))
|
||
|
|
self.assertEqual(f'\x1b[32m{strg}\x1b[39m', chalk.green(strg))
|
||
|
|
self.assertEqual(f'\x1b[34m{strg}\x1b[39m', chalk.blue(strg))
|
||
|
|
self.assertEqual(f'\x1b[36m{strg}\x1b[39m', chalk.cyan(strg))
|
||
|
|
self.assertEqual(f'\x1b[33m{strg}\x1b[39m', chalk.yellow(strg))
|
||
|
|
self.assertEqual(f'\x1b[35m{strg}\x1b[39m', chalk.purple(strg))
|
||
|
|
|
||
|
|
def test_chalk_bold(self):
|
||
|
|
strg = "The quick brown fox jumps over the lazy dog"
|
||
|
|
self.assertEqual(f'\x1b[1m{strg}\x1b[22m', chalk.bold(strg))
|
||
|
|
self.assertEqual(f'\x1b[2m{strg}\x1b[22m', chalk.faint(strg))
|
||
|
|
self.assertEqual(f'\x1b[1m\x1b[33m{strg}\x1b[39m\x1b[22m', chalk.bold.yellow(strg))
|