Приспособление Pytest, возвращающее кратные - PullRequest
0 голосов
/ 13 сентября 2018

Возможно ли иметь прибор, который возвращает словарь и фрейм данных?

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

Ответы [ 2 ]

0 голосов
/ 13 сентября 2018

Самый читаемый способ - вернуть namedtuple в приборе:

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
0 голосов
/ 13 сентября 2018

Возвращаемое значение (или выходное значение) прибора буквально является объектом, введенным в качестве аргумента функции во время выполнения теста:

def test_check(setup):
    dictionary, dataframe = setup
    assert dictionary['movie']['action'] == 'Avengers'
    assert dataframe.shape[0] == 5
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...