import allure
import pytest
import yaml

# from pytest_code.calc import Calculator


with open("./data_source/datasource.yaml") as f:
    data_source = yaml.safe_load(f)['datas']
    add_datas = data_source['add']
    print(add_datas)
    div_datas = data_source['div']
    print(div_datas)


@allure.feature("计算器加法和除法-测试用例")
class TestAddAndDiv:

    # def setup_class(self):
    #     print("开始计算")
    #     # 实例化计算器类
    #     self.calc = Calculator()
    #
    # def teardown_class(self):
    #     print("计算结束")

    @allure.title("测试相加_{a}_{b}_{expect}")
    @pytest.mark.parametrize('a,b,expect', add_datas)
    def test_add(self, get_calc, a, b, expect):
        # 调用 add 方法
        result = get_calc.add(a, b)
        # 若result1为float型，就保留两位小数
        if isinstance(result, float):
            result = round(result, 4)
        # 加法的断言
        assert result == expect

    @allure.title("测试相除_{a}_{b}_{expect}")
    @pytest.mark.parametrize('a,b,expect', div_datas)
    def test_div(self, get_calc, a, b, expect):
        if b == 0:
            print("分母为零，无法计算")
        else:
            # 调用 div 方法
            result = get_calc.div(a, b)
            # 若除数为0，则result记为null
            if isinstance(result, float):
                result = round(result, 4)
            # 除法的断言
            assert result == expect