dream

1
import requests
2

3

4
def main():
5
    url = "http://0.0.0.0:8136/respond"
6

7
    request_data = [{"utterances": [["i live in moscow"]]}]
8
    gold_results = [[{"triplets": [{"object": "moscow", "relation": "live in general", "subject": "user"}]}]]
9

10
    count = 0
11
    for data, gold_result in zip(request_data, gold_results):
12
        result = requests.post(url, json=data).json()
13
        if result and result[0] == gold_result:
14
            count += 1
15
        else:
16
            print(f"Got {result}, but expected: {gold_result}")
17
        print(result)
18

19
    assert count == len(request_data)
20
    print("Success")
21

22

23
if __name__ == "__main__":
24
    main()
25