def test_reset():
df = DialogueFlow(States.A, initial_speaker=Speaker.SYSTEM)
df.add_state(States.B, error_successor=States.E)
df.add_state(States.C, error_successor=States.E)
df.add_system_transition(States.A, States.B, "B")
df.add_system_transition(States.A, States.C, "C")
df.add_system_transition(States.E, States.E, "E")
# response = df.system_turn()
df.system_turn()
df.user_turn("")
assert df.state() == States.E
df.reset()
assert df.state() == "States.A"
assert df.speaker() == Speaker.SYSTEM
# response = df.system_turn()
df.system_turn()
df.user_turn("")
assert df.state() == States.E
df.system_turn()
assert df.state() == States.E
df.reset()
assert df.state() == States.A
assert df.speaker() == Speaker.SYSTEM
# response = df.system_turn()
df.system_turn()
df.user_turn("")
assert df.state() == States.E
df.system_turn()
assert df.state() == States.E
def test_reset():
df = DialogueFlow(States.A, initial_speaker=Speaker.SYSTEM)
df.add_state(States.B, error_successor=States.E)
df.add_state(States.C, error_successor=States.E)
df.add_system_transition(States.A, States.B, 'B')
df.add_system_transition(States.A, States.C, 'C')
df.add_system_transition(States.E, States.E, 'E')
response = df.system_turn()
df.user_turn("")
assert df.state() == States.E
df.reset()
assert df.state() == 'States.A'
assert df.speaker() == Speaker.SYSTEM
response = df.system_turn()
df.user_turn("")
assert df.state() == States.E
df.system_turn()
assert df.state() == States.E
df.reset()
assert df.state() == States.A
assert df.speaker() == Speaker.SYSTEM
response = df.system_turn()
df.user_turn("")
assert df.state() == States.E
df.system_turn()
assert df.state() == States.E
def test_constructor():
df = DialogueFlow(States.A)
assert df.state() == States.A
assert df.speaker() == Speaker.SYSTEM