Tests using pytest fixtures

test_sushi.py

from dataclasses import dataclass, field
from typing import List

import pytest


@dataclass(frozen=True)
class Ingredient:
    name: str
    plant_based: bool


@dataclass(frozen=True)
class Sushi:
    name: str
    ingredients: List[Ingredient] = field(default_factory=list)

    def __contains__(self, ingredient: Ingredient):
        return ingredient in self.ingredients


@pytest.fixture
def salmon():
    return Ingredient(name="Salmon", plant_based=False)


@pytest.fixture
def rice():
    return Ingredient(name="Rice", plant_based=True)


@pytest.fixture
def nori():
    return Ingredient(name="Nori", plant_based=True)


@pytest.fixture
def sake_nigiri(salmon, rice, nori):
    return Sushi(name="Sake Nigiri", ingredients=[salmon, rice, nori])


def test_sake_nigiri_contains_salmon(sake_nigiri, salmon):
    assert salmon in sake_nigiri


def test_sake_nigiri_contains_rice(sake_nigiri):
    assert rice in sake_nigiri


def test_sake_nigiri_contains_nori(sake_nigiri, nori):
    assert nori in sake_nigiri