testing_code_after_clean_up.py

@pytest.mark.django_db
def test_pizza_price_with_toppings():
    # Preparation phase
    pizza = create_pizza(number_of_toppings=2)

    # Calculating results
    actual_price = pizza.total_price
    expected_price = sum([t.price for t in pizza.toppings.all()])

    # Assertion
    assert actual_price == expected_price


@pytest.mark.django_db
def test_pizza_price_without_toppings():
    # Preparation phase
    pizza = create_pizza(number_of_toppings=0)

    # Calculating results
    actual_price = pizza.total_price
    expected_price = Decimal(0)

    # Assertion
    assert actual_price == expected_price