tests/test_name_signal_extractor.py

import pytest

from app.services.name_signal_extractor import extract_preferred_name


@pytest.mark.parametrize(
    ("text", "expected"),
    [
        ("My name is Anshuman.", "Anshuman"),
        ("My name is Anshuman, and I'm studying robotics.", "Anshuman"),
        ("My name is Anshuman and I'm studying robotics.", "Anshuman"),
        ("Please call me Anshuman.", "Anshuman"),
        ("You can call me Ash.", "Ash"),
        ("I go by Ash.", "Ash"),
        ("I am confused about robotics.", None),
        ("I am working on a robotics project.", None),
        ("I am not sure what to do next.", None),
    ],
)
def test_extract_preferred_name(text, expected):
    assert extract_preferred_name(text) == expected