File size: 953 Bytes
ae0863a
48f5500
 
 
 
c01e8a6
 
48f5500
 
 
 
 
ae0863a
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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, .."))