25 lines
671 B
Python
25 lines
671 B
Python
import pytest
|
|
|
|
from image_prediction.utils.process_wrapping import wrap_in_process
|
|
|
|
|
|
@pytest.fixture
|
|
def process_fn_mock():
|
|
def _process(a: str, b: float, c: dict, d: list):
|
|
return a * 2, b + 3, c, d[0]
|
|
|
|
return _process
|
|
|
|
|
|
@pytest.fixture
|
|
def parameter():
|
|
return {"a": "A", "b": 0.42, "c": {"x": 1, "y": 2}, "d": [1, 2, 3]}
|
|
|
|
|
|
def test_process_wrapper_with_args(process_fn_mock, parameter):
|
|
assert process_fn_mock(*parameter.values()) == wrap_in_process(process_fn_mock)(*parameter.values())
|
|
|
|
|
|
def test_process_wrapper_with_kwargs(process_fn_mock, parameter):
|
|
assert process_fn_mock(**parameter) == wrap_in_process(process_fn_mock)(**parameter)
|