def test_memorise_with_nlu(self, trained_policy: MemoizationPolicy, default_domain: Domain): tracker = tracker_from_dialogue(TEST_DEFAULT_DIALOGUE, default_domain) states = trained_policy._prediction_states(tracker, default_domain) recalled = trained_policy.recall(states, tracker, default_domain, None) assert recalled is not None
async def test_inmemory_tracker_store(pair): dialogue, domainpath = pair domain = Domain.load(domainpath) tracker = tracker_from_dialogue(dialogue, domain) tracker_store = InMemoryTrackerStore(domain) await tracker_store.save(tracker) restored = await tracker_store.retrieve(tracker.sender_id) assert restored == tracker
def moodbot_tracker(moodbot_domain: Domain) -> DialogueStateTracker: return tracker_from_dialogue(TEST_MOODBOT_DIALOGUE, moodbot_domain)
def test_dialogue_from_parameters(domain: Domain): tracker = tracker_from_dialogue(TEST_DEFAULT_DIALOGUE, domain) serialised_dialogue = InMemoryTrackerStore.serialise_tracker(tracker) deserialised_dialogue = Dialogue.from_parameters( json.loads(serialised_dialogue)) assert tracker.as_dialogue().as_dict() == deserialised_dialogue.as_dict()
def test_tracker_default(domain: Domain): tracker = tracker_from_dialogue(TEST_DEFAULT_DIALOGUE, domain) assert tracker.get_slot("name") == "Peter" assert tracker.get_slot("price") is None # slot doesn't exist!