py.test : Can multiple markers be applied at the test function level?

hpn picture hpn · Jul 12, 2016 · Viewed 10.2k times · Source

I have seen from the pytest docs that we can apply multiple markers at once on the Class or module level. I didn't find documentation for doing it at the test function level. Has anybody done this before with success?

I would like to ideally do this as a list of markers as being done in the above doc for Classes, for example (quoting from the docs):

class TestClass:
    pytestmark = [pytest.mark.webtest, pytest.mark.slowtest]

So, the pytest documentation talks about using pytestmark to specify the markers at the class and module level. However, it doesn't talk about having something similar at the test function level. I would have to specify the markers individually on top of test functions to get them marked with each one of them. This makes the test code look a little clunky with the increasing number of markers on top of test functions.

test_example.py:

pytestmark = [class1, class2]

class TestFeature(TestCase):

    @pytest.mark.marker1
    @pytest.mark.marker2
    @pytest.mark.marker3
    def test_function(self):
        assert True

Answer

Suor picture Suor · Aug 5, 2017

For functions you just repeat the decorator:

@pytest.mark.webtest
@pytest.mark.slowtest
def test_something(...):
    ...

If you want to reuse that for several tests you should remember that decorators are just functions returning decorated thing, so several decorators is just a composition:

def compose_decos(decos):
    def composition(func):
        for deco in reversed(decos):
            func = deco(func)
        return func
    return composition

all_marks = compose_decos(pytest.mark.webtest, pytest.mark.slowtest)

@all_marks
def test_something(...):
    ...

Or you can use general purpose composition such as my funcy library has:

from funcy import compose

all_marks = compose(pytest.mark.webtest, pytest.mark.slowtest)

Note that this way you can compose any decorators, not only pytest marks.