dream

1
import allure
2
import pytest
3
import requests
4

5

6
@allure.description("""Test fact retrieval""")
7
@pytest.mark.parametrize(
8
    "request_data, gold_results",
9
    [
10
        (
11
            {
12
                "dialog_history": [["Какая столица России?"]],
13
                "entity_substr": [["россии"]],
14
                "entity_tags": [["loc"]],
15
                "entity_pages": [[["Россия"]]],
16
            },
17
            "Росси́я или Росси́йская Федера́ция (РФ), — государство в Восточной Европе"
18
            " и Северной Азии. Территория России"
19
            " в её конституционных границах составляет км²; население страны (в пределах её заявленной территории) "
20
            "составляет чел. (). Занимает первое место в мире по территории, шестое — по объёму ВВП по ППС, и девятое "
21
            "— по численности населения. Столица — Москва. Государственный язык — русский. Денежная единица — "
22
            "российский рубль.",
23
        )
24
    ],
25
)
26
def test_fact_retrieval_rus(url: str, request_data: dict, gold_results: str):
27
    response = requests.post(url, json=request_data)
28
    result = response.json()
29
    assert response.status_code == 200
30
    assert result[0] and result[0][0] and result[0][0][0] == gold_results
31