chat/test_sanitize.py

from chat import sanitize, target


def test_sanitize():
    assert '' == sanitize('')
    assert '' == sanitize('((a)')
    assert '(' == sanitize('(')
    assert 'qwertyui' == sanitize('qwertyui')
    assert 'qwertyui ' == sanitize('qwertyui (abcdef)')
    assert 'qwertyui  poiuy' == sanitize('qwertyui (abcdef) poiuy')
    assert 'qwertyui  poiuy' == sanitize('Priest says: qwertyui (abcdef) poiuy')

    assert '"Teacher, I heard about a fascinating story from someone who lives in the city. Can you guess who it might be?" ' == sanitize(
        '"Teacher, I heard about a fascinating story from someone who lives in the city. Can you guess who it might be?" (I want to find out if the Teacher can make any...)')


def test_target():
    assert 'Priest' == target("Priest, your task is to figure out their names and where they live. Do not ask directly, they must not realize what information you are after!")
    assert 'Priest' == target(sanitize("(I'm thinking hmm) Priest, .."))