test_sanitize.py

from app 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 '"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 connections between my statement and their own information.)')

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, .."))