dream

1
import requests
2
import json
3

4

5
def get_input_json(fname):
6
    with open(fname, "r") as f:
7
        res = json.load(f)
8
    return {"dialogs": [res]}
9

10

11
def test_one_step_responses():
12
    url = "http://0.0.0.0:8091/model"
13

14
    print("test annotations")
15
    input_data = get_input_json("test_configs/test_annotations.json")
16
    response = requests.post(url, json=input_data).json()[0]
17
    assert response[0], print(response)
18

19
    print("test no annotations")
20
    input_data = get_input_json("test_configs/test_no_annotations.json")
21
    response = requests.post(url, json=input_data).json()[0]
22
    assert response[0], print(response)
23

24
    print("SUCCESS!")
25

26

27
if __name__ == "__main__":
28
    test_one_step_responses()
29