File size: 989 Bytes
c32815c
48f5500
 
 
 
c01e8a6
 
48f5500
 
 
c32815c
48f5500
4577a87
 
 
ae0863a
 
 
4577a87
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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, .."))