def test_colored():
    ret = color('foo', CYAN, True)
    assert ret == CYAN + 'foo' + NORMAL
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'
def test_color_no_color():
    ret = color('foo', 'bar', False)
    assert ret == 'foo'