Cenário 1 do artigo: https://elyssonmr.com/blog/posts/20241231_tests_hints/

database.py

def save_pokemon_name(pokemon_name: str):
    # Should save pokemon name in database
    pass

helpers.py

from scenario1.database import save_pokemon_name


def save_pokemons_name(pokemons_names: list[str]):
    for pokemon_name in pokemons_names:
        save_pokemon_name(pokemon_name=pokemon_name)

test_save_db.py

from unittest.mock import patch

import pytest

from scenario1.helpers import save_pokemons_name


@pytest.fixture
def mock_save_pokemon_name():
    with patch('scenario1.helpers.save_pokemon_name') as patched:
        yield patched


def test_should_save_pokemons(mock_save_pokemon_name):
    pokemons_names = ['Bulbasaur']

    save_pokemons_name(pokemons_names)

    mock_save_pokemon_name.assert_called_once_with(
        pokemon_name=pokemons_names[0]
    )