tests/test_explore.py

from math_jlens.explore import (
    extract_boxed,
    find_last_subsequence,
    normalize_answer,
    response_answer,
)


def test_extract_balanced_boxed_answer():
    text = r"Work. Answer: \boxed{\frac{3}{7}}"
    assert extract_boxed(text) == r"\frac{3}{7}"
    assert response_answer(text) == r"\frac{3}{7}"


def test_answer_normalization():
    assert normalize_answer(" $1,234$ ") == "1234"


def test_find_last_subsequence():
    assert find_last_subsequence([1, 2, 3, 2, 3], [2, 3]) == 3
    assert find_last_subsequence([1, 2], [3]) is None