tests/test_trace_explorer.py

from __future__ import annotations

import pandas as pd

from src.text_search import literal_text_mask


def test_literal_text_mask_handles_regex_metacharacters() -> None:
    values = pd.Series(["What is [risk] exposure?", "Explain risk exposure", "No match"])

    mask = literal_text_mask(values, "[risk]")

    assert mask.tolist() == [True, False, False]


def test_literal_text_mask_is_case_insensitive() -> None:
    values = pd.Series(["Financial REPORT risk", "clinical note", None])

    mask = literal_text_mask(values, "report")

    assert mask.tolist() == [True, False, False]