Пример #1
0
    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
Пример #2
0
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
Пример #3
0
def moodbot_tracker(moodbot_domain: Domain) -> DialogueStateTracker:
    return tracker_from_dialogue(TEST_MOODBOT_DIALOGUE, moodbot_domain)
Пример #4
0
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()
Пример #5
0
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!