def test_colored(): ret = color('foo', CYAN, True) assert ret == CYAN + 'foo' + NORMAL
def test_color_no_color(): ret = color('foo', 'bar', False) assert ret == 'foo'