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_unexpected_input_macro():
df = DialogueFlow('root', initial_speaker=Speaker.USER)
df.add_user_transition('root', 'x', '#UNX')
df.add_system_transition('x', 'y', '"So, whats for dinner?"')
df.user_turn('blah blah blah blah blah blah')
tokens = df.system_turn().split()
assert len(tokens) > 4 and 'So, whats for dinner?' == ' '.join(tokens[-4:])
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_unexpected_input_macro():
df = DialogueFlow("root", initial_speaker=Speaker.USER)
df.add_user_transition("root", "x", "#UNX")
df.add_system_transition("x", "y", '"So, whats for dinner?"')
df.user_turn("blah blah blah blah blah blah")
tokens = df.system_turn().split()
assert len(tokens) > 4 and "So, whats for dinner?" == " ".join(tokens[-4:])
def test_global_transition_list_disjunction():
df = DialogueFlow("root", initial_speaker=Speaker.USER)
nlu = ["hi", "hello"]
df.add_state("root", "x")
df.add_system_transition("x", "y", "hello")
df.add_global_nlu("x", nlu)
df.user_turn("ok")
df.system_turn()
df.user_turn("hi")
assert df.state() == "x"
def test_ontology():
kb = KnowledgeBase()
ontology = {
"ontology": {
"season": [
"fall",
"spring",
"summer",
"winter"
],
"month": [
"january",
"february",
"march",
"april",
"may",
"june",
"july",
"august",
"september",
"october",
"november",
"december"
]
}
}
kb.load_json(ontology)
df = DialogueFlow(States.A, Speaker.USER, kb=kb)
df.add_state(States.A)
df.add_state(States.B)
df.add_state(States.C)
df.add_state(States.D)
df.add_state(States.E)
df.set_error_successor(States.A, States.E)
df.set_error_successor(States.B, States.E)
df.set_error_successor(States.C, States.E)
df.set_error_successor(States.D, States.E)
df.add_user_transition(States.A, States.B, "[#ONT(month)]")
df.add_system_transition(States.B, States.C, "B to C")
df.add_user_transition(States.C, States.D, "[$m=#ONT(month), $s=#ONT(season)]")
df.user_turn("january")
assert df.state() == States.B
assert df.system_turn() == "B to C"
df.user_turn("october is in the fall season")
assert df.state() == States.D
assert df._vars["m"] == "october"
assert df._vars["s"] == "fall"
df.set_state(States.A)
df.set_speaker(Speaker.USER)
df.user_turn("hello there", debugging=False)
assert df.state() == States.E
def test_global_transition_list_disjunction():
df = DialogueFlow('root', initial_speaker=Speaker.USER)
nlu = [
'hi',
'hello'
]
df.add_state('root', 'x')
df.add_system_transition('x', 'y', 'hello')
df.add_global_nlu('x', nlu)
df.user_turn('ok')
df.system_turn()
df.user_turn('hi')
assert df.state() == 'x'
def test_add_transitions():
df = DialogueFlow("States.A")
df.add_system_transition("States.A", "States.B", "hello")
df.add_user_transition("States.B", "States.C", "[{hi, hello, hey, [how, you]}]")
df.add_user_transition("States.B", "States.D", "[{dog, cat, parrot}]")
df.add_system_transition("States.D", "States.A", "so")
assert df.graph().arcs() == {
("States.A", "States.B", DialogueFlow.Speaker.SYSTEM),
("States.B", "States.C", DialogueFlow.Speaker.USER),
("States.B", "States.D", DialogueFlow.Speaker.USER),
("States.D", "States.A", DialogueFlow.Speaker.SYSTEM),
}
assert isinstance(df.transition_natex("States.A", "States.B", Speaker.SYSTEM), NatexNLG)
assert isinstance(df.transition_natex("States.B", "States.C", Speaker.USER), NatexNLU)
def test_add_transitions():
df = DialogueFlow('States.A')
df.add_system_transition('States.A', 'States.B', 'hello')
df.add_user_transition('States.B', 'States.C', '[{hi, hello, hey, [how, you]}]')
df.add_user_transition('States.B', 'States.D', '[{dog, cat, parrot}]')
df.add_system_transition('States.D', 'States.A', 'so')
assert df.graph().arcs() == {
('States.A', 'States.B', DialogueFlow.Speaker.SYSTEM),
('States.B', 'States.C', DialogueFlow.Speaker.USER),
('States.B', 'States.D', DialogueFlow.Speaker.USER),
('States.D', 'States.A', DialogueFlow.Speaker.SYSTEM)
}
assert isinstance(df.transition_natex('States.A', 'States.B', Speaker.SYSTEM), NatexNLG)
assert isinstance(df.transition_natex('States.B', 'States.C', Speaker.USER), NatexNLU)
def test_system_transition():
for _ in range(6):
df = DialogueFlow(States.A)
df.add_system_transition(States.A, States.B, "hello", score=2.0)
df.add_system_transition(States.A, States.C, "hey")
assert df.system_transition(df.state()) == ("hello", States.B)
df = DialogueFlow(States.A)
df.add_system_transition(States.A, States.B, "hello")
df.add_system_transition(States.A, States.C, "hey", score=2.0)
assert df.system_transition(df.state()) == ("hey", States.C)
def test_system_transition():
for _ in range(6):
df = DialogueFlow(States.A)
df.add_system_transition(States.A, States.B, 'hello', score=2.0)
df.add_system_transition(States.A, States.C, 'hey')
assert df.system_transition(df.state()) == ('hello', States.B)
df = DialogueFlow(States.A)
df.add_system_transition(States.A, States.B, 'hello')
df.add_system_transition(States.A, States.C, 'hey', score=2.0)
assert df.system_transition(df.state()) == ('hey', States.C)
def test_information_state_based_dialogue():
df = DialogueFlow("root", initial_speaker=Speaker.USER)
df.add_user_transition("root", "one", "$x=[{cat, dog}]")
df.set_error_successor("root", "one")
df.add_system_transition("one", "root", "i am a stupid state machine")
df.add_update_rule("[$x={ice cream, candy}]")
df.add_update_rule("#ANY($x=candy)", "$y=blah")
df.add_update_rule("#ANY($x=ice cream)", "$y=None")
df.add_update_rule("#ANY($y=blah)", "i am a smart info state manager (2.0)")
df.user_turn("candy", debugging=True)
assert df.system_turn().strip() == "i am a smart info state manager"
df.user_turn("ice cream", debugging=True)
assert df.system_turn(debugging=True).strip() == "i am a stupid state machine"
df.user_turn("ice cream", debugging=True)
assert df.system_turn(debugging=True).strip() == "i am a stupid state machine"
df.user_turn("ice cream", debugging=True)
assert df.system_turn(debugging=True).strip() == "i am a stupid state machine"
df.user_turn("ice cream", debugging=True)
assert df.system_turn(debugging=True).strip() == "i am a stupid state machine"
def test_information_state_based_dialogue():
df = DialogueFlow('root', initial_speaker=Speaker.USER)
df.add_user_transition('root', 'one', '$x=[{cat, dog}]')
df.set_error_successor('root', 'one')
df.add_system_transition('one', 'root', 'i am a stupid state machine')
df.add_update_rule('[$x={ice cream, candy}]')
df.add_update_rule('#ANY($x=candy)', '$y=blah')
df.add_update_rule('#ANY($x=ice cream)', '$y=None')
df.add_update_rule('#ANY($y=blah)', 'i am a smart info state manager (2.0)')
df.user_turn('candy', debugging=True)
assert df.system_turn().strip() == 'i am a smart info state manager'
df.user_turn('ice cream', debugging=True)
assert df.system_turn(debugging=True).strip() == 'i am a stupid state machine'
df.user_turn('ice cream', debugging=True)
assert df.system_turn(debugging=True).strip() == 'i am a stupid state machine'
df.user_turn('ice cream', debugging=True)
assert df.system_turn(debugging=True).strip() == 'i am a stupid state machine'
df.user_turn('ice cream', debugging=True)
assert df.system_turn(debugging=True).strip() == 'i am a stupid state machine'
def test_system_multi_hop():
df = DialogueFlow(States.A, initial_speaker=Speaker.SYSTEM)
df.add_state(States.B, error_successor=States.B, system_multi_hop=True)
df.add_state(States.C, error_successor=States.A, system_multi_hop=True)
df.add_state(States.D, error_successor=States.A)
df.add_state(States.E, error_successor=States.A)
df.add_system_transition(States.A, States.B, "{hey, hello}")
df.add_system_transition(States.A, States.C, "excuse me")
df.add_system_transition(States.B, States.D, "how are you")
df.add_system_transition(States.C, States.E, "what")
for _ in range(100):
df.set_state(States.A)
df.set_speaker(Speaker.SYSTEM)
response = df.system_turn()
assert response in {"hey how are you", "hello how are you", "excuse me what"}
def test_system_multi_hop():
df = DialogueFlow(States.A, initial_speaker=Speaker.SYSTEM)
df.add_state(States.B, error_successor=States.B, system_multi_hop=True)
df.add_state(States.C, error_successor=States.A, system_multi_hop=True)
df.add_state(States.D, error_successor=States.A)
df.add_state(States.E, error_successor=States.A)
df.add_system_transition(States.A, States.B, '{hey, hello}')
df.add_system_transition(States.A, States.C, 'excuse me')
df.add_system_transition(States.B, States.D, 'how are you')
df.add_system_transition(States.C, States.E, 'what')
for _ in range(100):
df.set_state(States.A)
df.set_speaker(Speaker.SYSTEM)
response = df.system_turn()
assert response in {'hey how are you', 'hello how are you',
'excuse me what'}
("also", "type", "also_syns"),
("too", "type", "also_syns"),
("basketball", "type", "unknown_hobby"),
("basketball", "expr", "bball"),
("john", "type", "male"),
("mary", "type", "female"),
("male", "type", "person"),
("female", "type", "person"),
("friend", "type", "person"),
])
ont = {"ontology": {"person": ["mom", "dad", "sister", "brother"]}}
kb.load_json(ont)
df = DialogueFlow(initial_state="start")
df.add_state("start", "start")
df.add_system_transition("start", "start", "hello")
macros = {
# "ONT": ONTE(kb),
# "KBQ": KBE(kb),
# "EXP": EXP(kb),
# "U": UnionMacro(),
# "I": Intersection(),
# "DIF": Difference(),
# "SET": SetVars(),
# "ALL": CheckVarsConjunction(),
# "ANY": CheckVarsDisjunction(),
# "NOT": NOT(),
# "ISP": IsPlural(),
# "FPP": FirstPersonPronoun(kb),
# "PSP": PossessivePronoun(kb),
def test_check():
df = DialogueFlow(States.A)
df.add_state(States.B, error_successor=States.C)
df.add_state(States.D, error_successor=States.A)
df.add_system_transition(States.A, States.B, "hello")
df.add_user_transition(States.B, States.C, "[{hi, hello, hey, [how, you]}]")
df.add_user_transition(States.B, States.A, "[{bye, goodbye, see you, see ya, later}]")
df.add_system_transition(States.C, States.D, "ok")
assert df.check()
df = DialogueFlow(States.A)
df.add_state(States.B, error_successor=States.C)
df.add_system_transition(States.A, States.B, "hello")
df.add_user_transition(States.B, States.C, "[{hi, hello, hey, [how, you]}]")
df.add_user_transition(States.B, States.A, "[{bye, goodbye, see you, see ya, later}]")
df.add_system_transition(States.C, States.D, "ok")
assert not df.check()
df = DialogueFlow(States.A)
df.add_state(States.B, error_successor=States.C)
df.add_state(States.D, error_successor=States.A)
df.add_system_transition(States.A, States.B, "hello")
df.add_user_transition(States.B, States.C, "[{hi, hello, hey, [how, you]}]")
df.add_user_transition(States.B, States.A, "[{bye, goodbye, see you, see ya, later}]")
df.add_system_transition(States.C, States.D, "ok then $X alright")
assert not df.check()
def test_serialization():
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, '#GATE $spoken=B')
df.add_system_transition(States.A, States.C, '#GATE(spoken) $spoken=C')
df.add_system_transition(States.E, States.E, '$spoken=E')
df.add_user_transition(States.B, States.A, '$heard=b')
df.add_user_transition(States.E, States.A, '$heard=e')
df.system_turn()
assert df.state() == States.B
assert df.vars()["spoken"] == "B"
df.user_turn('b')
df.system_turn()
assert df.state() == States.C
assert df.vars()["spoken"] == "C"
expected_gates = {'States.B': [{}], 'States.C': [{'spoken': 'B'}]}
assert df.gates() == expected_gates
df.vars()["testing_none"] = None
d = df.serialize()
df2 = DialogueFlow(States.A, initial_speaker=Speaker.SYSTEM)
df2.add_state(States.B, error_successor=States.E)
df2.add_state(States.C, error_successor=States.E)
df2.add_system_transition(States.A, States.B, '#GATE $spoken=B')
df2.add_system_transition(States.A, States.C, '#GATE(spoken) $spoken=C')
df2.add_system_transition(States.E, States.E, '$spoken=E')
df2.add_user_transition(States.B, States.A, '$heard=b')
df2.add_user_transition(States.E, States.A, '$heard=e')
assert df2.state() == States.A
assert 'spoken' not in df2.vars() and 'heard' not in df2.vars()
assert len(df2.gates()) == 0
df2.deserialize(d)
assert df.vars() == df2.vars()
assert df.gates() == df2.gates()
assert df.state() == df2.state()
assert df2.vars()["testing_none"] is None
('lion', 'type', 'cat'), ('panther', 'type', 'cat'),
('cat', 'type', 'animal'), ('animal', 'type', 'thing'),
('lion', 'sound', 'roar'), ('roar', 'quality', 'loud'),
('panther', 'sound', 'growl'), ('growl', 'quality', 'scary'),
('also', 'type', 'also_syns'), ('too', 'type', 'also_syns'),
('basketball', 'type', 'unknown_hobby'), ('basketball', 'expr', 'bball'),
('john', 'type', 'male'), ('mary', 'type', 'female'),
('male', 'type', 'person'), ('female', 'type', 'person'),
('friend', 'type', 'person')
])
ont = {"ontology": {"person": ["mom", "dad", "sister", "brother"]}}
kb.load_json(ont)
df = DialogueFlow(initial_state='start')
df.add_state('start', 'start')
df.add_system_transition('start', 'start', 'hello')
macros = {
'ONT': ONTE(kb),
'KBQ': KBE(kb),
'EXP': EXP(kb),
'U': UnionMacro(),
'I': Intersection(),
'DIF': Difference(),
'SET': SetVars(),
'ALL': CheckVarsConjunction(),
'ANY': CheckVarsDisjunction(),
'NOT': NOT(),
'ISP': IsPlural(),
'FPP': FirstPersonPronoun(kb),
'PSP': PossessivePronoun(kb),
def test_single_system_transition():
df = DialogueFlow(States.A)
df.add_system_transition(States.A, States.B, "hello")
assert df.system_transition(df.state()) == ("hello", States.B)
def test_serialization():
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, "#GATE $spoken=B")
df.add_system_transition(States.A, States.C, "#GATE(spoken) $spoken=C")
df.add_system_transition(States.E, States.E, "$spoken=E")
df.add_user_transition(States.B, States.A, "$heard=b")
df.add_user_transition(States.E, States.A, "$heard=e")
df.system_turn()
assert df.state() == States.B
assert df.vars()["spoken"] == "B"
df.user_turn("b")
df.system_turn()
assert df.state() == States.C
assert df.vars()["spoken"] == "C"
expected_gates = {"States.B": [{}], "States.C": [{"spoken": "B"}]}
assert df.gates() == expected_gates
df.vars()["testing_none"] = None
d = df.serialize()
df2 = DialogueFlow(States.A, initial_speaker=Speaker.SYSTEM)
df2.add_state(States.B, error_successor=States.E)
df2.add_state(States.C, error_successor=States.E)
df2.add_system_transition(States.A, States.B, "#GATE $spoken=B")
df2.add_system_transition(States.A, States.C, "#GATE(spoken) $spoken=C")
df2.add_system_transition(States.E, States.E, "$spoken=E")
df2.add_user_transition(States.B, States.A, "$heard=b")
df2.add_user_transition(States.E, States.A, "$heard=e")
assert df2.state() == States.A
assert "spoken" not in df2.vars() and "heard" not in df2.vars()
assert len(df2.gates()) == 0
df2.deserialize(d)
assert df.vars() == df2.vars()
assert df.gates() == df2.gates()
assert df.state() == df2.state()
assert df2.vars()["testing_none"] is None
