diff --git a/src/test/dialog_state_monitor.py b/src/test/dialog_state_monitor.py index 5e3b5d2..ae4b44a 100644 --- a/src/test/dialog_state_monitor.py +++ b/src/test/dialog_state_monitor.py @@ -5,6 +5,7 @@ from src.model.slot import Slot dsm = DialogStateMonitor() assert dsm.pizza_exists('capri') is True +assert dsm.pizza_exists('buraczana') is False assert dsm.state['was_previous_order_invalid'] is False frame1 = Frame('user', 'inform/order', [Slot('pizza', 'margarita'), Slot('sauce', 'ketchup')]) @@ -42,7 +43,7 @@ assert len(dsm.state['history']) == 0 dsm.reset() -frame1 = Frame('user', 'inform/order', [Slot('pizza', 'buraczna')]) +frame1 = Frame('user', 'inform/order', [Slot('pizza', 'buraczana')]) dsm.update(frame1) assert dsm.state['was_previous_order_invalid'] is True assert dsm.state['belief_state']['order'] == []