from rainbow_parenthesis import colorize, Colored
import io


def test_escaped_parenthesis():
    assert list(r"\(") == list(colorize(io.StringIO(r"\(")))


def test_escaped_parenthesis_in_string():
    assert list(r"'\('") == list(colorize(io.StringIO(r"'\('")))


def test_escaped_quot_in_string():
    assert list(r"'\''") == list(colorize(io.StringIO(r"'\''")))


def test_embedded_string():
    assert list(colorize(io.StringIO("( '(' )"))) == [
        Colored(depth=0, item='('),
        *" '(' ",
        Colored(depth=0, item=')'),
    ]