Parameterize a pytest with functions

user_5208867 0
Viewed 423

I have a number of different custom transformer functions, and want to run a simple test (e.g. printing out the transformer) on all of these functions. I could write individual tests for each function, but thought there must be an easier way.

I've looked at parameterizing the test but get: TypeError: 'function' object is not iterable

@pytest.fixture
def list_transformers(self):
    list_transformers = [
        TransformerOne(column='a'),
        TransformerTwo(column='a')
    ]
    return list_trasformers

@pytest.mark.parametrize("transformer", list_transformers)
    def test_print(self, transformer):
        tf_type = transformer[0]
        params = transformer[1]
        tf = tf_type(**params)
        print(tf)

I think this is because the transformer functions are not returning something iterable. Should I be going about this in a different way?

pytest python
3 Answers

Why don't you just pass the list to the decorator:

@pytest.mark.parametrize("transformer", [TransformerOne(column='a'), TransformerTwo(column='a')])
def test_print(self, transformer):
    ...
user_63375230

Using fixtures in pytest.mark.parametrize is not yet fully supported.

Instead of using @pytest.mark.parametrize, you can use another style of parametrized tests which is via pytest_generate_tests.

import pytest


def pytest_generate_tests(metafunc):
    if "transformer" in metafunc.fixturenames:
        list_transformers = [
            'This is a',
            'While that is b',
        ]
        metafunc.parametrize("transformer", list_transformers)


def test_print(transformer):
    print(transformer)

Output:

$ pytest -q -rP
..                                                                                                                                                                                                  [100%]
================================================================================================= PASSES ==================================================================================================
__________________________________________________________________________________________ test_print[This is a] __________________________________________________________________________________________
------------------------------------------------------------------------------------------ Captured stdout call -------------------------------------------------------------------------------------------
This is a
_______________________________________________________________________________________ test_print[While that is b] _______________________________________________________________________________________
------------------------------------------------------------------------------------------ Captured stdout call -------------------------------------------------------------------------------------------
While that is b
2 passed in 0.07s
user_110438250

The second parameter of pytest.mark.parametrize() have to be an iterator not a function reference.

Something like this: @pytest.mark.parametrize("transformer", list_transformers())

or @pytest.mark.parametrize("transformer", [TransformerOne(column='a'), TransformerTwo(column='a')])

user_112063080
Related