从夹具返回结构化数据的正确方法是什么？

Question

目前我有这样的设置：

./
  src/
    mylib.py
  tests/
    __init__.py
    conftest.py
    test_mylib.py

mylib.py包含：

def add(x, y):
    return x + y

conftest.py包含：

import attr
import pytest

@attr.s(slots=True)
class AdditionTestCase:
    input1 = attr.ib()
    input2 = attr.ib()
    output_expected = attr.ib()

@pytest.fixture
def addition_test_case() -> AdditionTestCase:
    # make up some data
    return AdditionTestCase(1, 2, 3)

并且test_mylib.py包含：

from mylib import add
from tests.conftest import AdditionTestCase

def test_add(addition_test_case: AdditionTestCase):
    input1 = addition_test_case.input1
    input2 = addition_test_case.input2
    output_expected = addition_test_case.output_expected

    output_actual = add(input1, input2)
    assert output_actual == output_expected

这有什么问题吗？ 有更好的方法吗？ 如果从它导入被认为是不好的做法，我可以将类型定义conftest.py之外。

Answer 1

通常这可以通过仅使用没有额外类型的参数化来解决

@pytest.mark.parametrize("left, right, expected_output", [
   (1, 1, 2),
   (2, 2, 4),
]
def test_add(left, right, expected_output):
  output = add(left, right)
  assert output == expected_output