from bopytex.jinja2_env.texenv import texenv


def test_variable_block():
    base_template = r"\Var{a}"
    jinja2_template = texenv.from_string(base_template)
    output = jinja2_template.render(a=2)
    assert output == "2"


def test_block_string():
    base_template = r"\Block{set a = 2}\Var{a}"
    jinja2_template = texenv.from_string(base_template)
    output = jinja2_template.render()
    assert output == "2"


def test_block_line_statement():
    base_template = r"""%-set a = 2
\Var{a}"""
    jinja2_template = texenv.from_string(base_template)
    output = jinja2_template.render()
    assert output == "2"

def test_block_line_statement_with_comment():
    base_template = r"""%-set a = 2
%# comment
\Var{a}"""
    jinja2_template = texenv.from_string(base_template)
    output = jinja2_template.render()
    assert output == "\n2"