import pytest
from Utils import yaml_handler
from Utils import req_handler
from TestData.automation.ai_test_data import AiTestData

class TestAi:

    yaml_path = r'automation/ai.yaml'
    yaml_data = yaml_handler.yaml_handler.get_case(yaml_path)
    @pytest.mark.flaky(reruns=AiTestData.rerun, reruns_delay=AiTestData.rerun_delay)
    @pytest.mark.parametrize('case',yaml_data)
    def test_ai(self,case):
        req_handler.ReqHandler.send_requests(case=case,var_class=AiTestData)