Pytest fixture returning multiples

user_9632267 0
Viewed 4301

Is it possible to have a fixture that returns a dictionary and dataframe?

import somefile
import pytest

@pytest.fixture()
def setup():
    dictionary, dataframe = somefile.get_Di_And_Df()
    return(dictionary, dataframe)

def test_check(setup):
    assert dictionary['movie']['action'] == 'Avengers'
    assert dataframe.shape[0] == 5
pytest python
3 Answers

The return value (or yield value) of the fixture is literally the object injected as a function argument during test execution:

def test_check(setup):
    dictionary, dataframe = setup
    assert dictionary['movie']['action'] == 'Avengers'
    assert dataframe.shape[0] == 5
user_6740390

The most readable way is to return a namedtuple in the fixture:

from collections import namedtuple

import somefile
import pytest

@pytest.fixture()
def setup():
    dictionary, dataframe = somefile.get_Di_And_Df()
    return namedtuple("Setup", "dictionary dataframe")(dictionary, dataframe)


def test_check(setup):
    assert setup.dictionary['movie']['action'] == 'Avengers'
    assert setup.dataframe.shape[0] == 5
user_2338130

you could mock the called method

@patch("somefile.get_Di_And_Df", MagicMock=(return_value=(dictionary, dataframe)))
def test_check():
  _dict, _df = somefile.get_Di_And_Df()
  assert _dict['movie']['action'] == 'Avengers'
  assert _df.shape[0] == 5
user_103153550
Related