Проверка вызова пропатченной функции

@DonFerro · Регистрация: 10.07.2022

Студворк — интернет-сервис помощи студентам

Добрый день!

Разбираюсь с использованием unittest.mock.patch

Пример тестируемого кода:

Python
1
2
3
4
5
6
7
8
# example.py
 
def find_next_prime(number):
    index = number
    while True:
        index += 1
        if is_prime(index):
            return index

Тест для pytest:

Python
1
2
3
4
5
6
7
8
9
# test_example.py
 
from unittest import mock
from trypytest.example import *
 
@mock.patch('trypytest.example.is_prime', name='mock_is_prime', return_value=True)
def test_find_next_prime_patched(mock_is_prime):
    assert find_next_prime(5) == 6
    assert mock_is_prime.assert_called_once_with(6)

Результат теста:

Code
1
2
3
4
5
6
7
8
9
10
11
mock_is_prime = <MagicMock name='mock_is_prime' id='2442652853888'>
 
    @mock.patch('trypytest.example.is_prime', name='mock_is_prime', return_value=True)
    def test_find_next_prime_patched(mock_is_prime):
        assert find_next_prime(5) == 6
>       assert mock_is_prime.assert_called_once_with(6)
E       AssertionError: assert None
E        +  where None = <bound method NonCallableMock.assert_called_once_with of <MagicMock name='mock_is_prime' id='2442652853888'>>(6)
E        +    where <bound method NonCallableMock.assert_called_once_with of <MagicMock name='mock_is_prime' id='2442652853888'>> = <MagicMock name='mock_is_prime' id='2442652853888'>.assert_called_once_with
 
tests\test_example.py:31: AssertionError

Как проверить, что is_prime был вызван однократно с аргументом 6?

@Welemir1 · 03.08.2022, 14:36

DonFerro, в юниттесте не пишут ассерты так как ты, там self.assertTrue например, почитай документацию по ассертУанс он как раз все проверил уже, а ты свои ассертом еще проверяешь его возвращаемое значение а оно конечно Нан, потому падает
так и написано в ошибке

Сообщение от DonFerro

AssertionError: assert None

убери ассерты и почитай как юниттест использует ассерты

@DonFerro · 03.08.2022, 15:00 **[ТС]**

Поправил, теперь работает, спасибо!

Python
1
2
3
4
@mock.patch('trypytest.example.is_prime', name='mock_is_prime', return_value=True)
def test_find_next_prime_patched(mock_is_prime):
    assert find_next_prime(5) == 6
    mock_is_prime.assert_called_once_with(6)

@Welemir1 · 03.08.2022, 17:07

Сообщение от DonFerro

assert find_next_prime(5) == 6

исправить! ))

@DonFerro · 03.08.2022, 22:30 **[ТС]**

А что не так с этим ассертом?

Добавлено через 26 минут
Если его убрать, то тест ничего не проверяет:

В обоих случаях

Code
1
find_next_prime(5) == 6

и

Code
1
find_next_prime(5) == 7

тест проходит, но второй вариант неправильный.
А вариант

Code
1
assert find_next_prime(5) == 7

дает ошибку:

Code
1
2
3
4
5
6
7
8
9
mock_is_prime = <MagicMock name='mock_is_prime' id='2565651410144'>
 
    @mock.patch('trypytest.example.is_prime', name='mock_is_prime', return_value=True)
    def test_find_next_prime_patched(mock_is_prime):
>       assert find_next_prime(5) == 7
E       assert 6 == 7
E        +  where 6 = find_next_prime(5)
 
tests\test_example.py:31: AssertionError

@Welemir1 · 04.08.2022, 06:07

Сообщение от DonFerro

А что не так с этим ассертом?

то есть ты не стал читать ничего из мной предложенного? и даже проигнорировал вот это

Сообщение от Welemir1

в юниттесте не пишут ассерты так как ты, там self.assertTrue например

у тебя не должно быть оператора assert в юнит тестах, у него свои ассерты, кастомные, как это еще объяснить?
у тебя должно быть

Python
1
self.assertEqual(find_next_prime(5), 6)

@DonFerro · 04.08.2022, 07:10 **[ТС]**

Я пишу тесты для pytest, а не для unittest.

@Welemir1 · 04.08.2022, 09:00

DonFerro, а, соррян, сбило с толку импорт мока, тогда да ассерты в питесте обычные(с виду). Беру свои слова назад, ничего менять больше не нужно

@DonFerro · 04.08.2022, 09:43 **[ТС]**

Welemir1, я правильно понимаю, что в pytest нет собственной реализации mock и нужно пользоваться сторонней, например от unittest?

@Welemir1 · 04.08.2022, 10:31

DonFerro, да, у них нет мока, но есть манкипатч, он тоже предоставляет возможности подмен, посмотри на него.

Новые блоги и статьи Все статьи Все блоги /
Использование SDL3-callbacks вместо функции main() на Android, Desktop и WebAssembly 8Observer8 24.01.2026 Если вы откроете примеры для начинающих на официальном репозитории SDL3 в папке: examples, то вы увидите, что все примеры используют следующие четыре обязательные функции, а привычная функция main(). . .	моя боль iceja 24.01.2026 Выложила интерполяцию кубическими сплайнами www. iceja. net REST сервисы временно не работают, только через Web. Написала за 56 рабочих часов этот сайт с нуля. При помощи perplexity. ai PRO , при. . .	Модель сукцессии микоризы anaschu 24.01.2026 Решили писать научную статью с неким РОманом	http://iceja.net/ математические сервисы iceja 20.01.2026 Обновила свой сайт http:/ / iceja. net/ , приделала Fast Fourier Transform экстраполяцию сигналов. Однако предсказывает далеко не каждый сигнал (см ограничения http:/ / iceja. net/ fourier/ docs ). Также. . .
http://iceja.net/ сервер решения полиномов iceja 18.01.2026 Выкатила http:/ / iceja. net/ сервер решения полиномов (находит действительные корни полиномов методом Штурма). На сайте документация по API, но скажу прямо VPS слабенький и 200 000 полиномов. . .	Расчёт переходных процессов в цепи постоянного тока igorrr37 16.01.2026 / * Дана цепь(не выше 3-го порядка) постоянного тока с элементами R, L, C, k(ключ), U, E, J. Программа находит переходные токи и напряжения на элементах схемы классическим методом(1 и 2 з-ны. . .	Восстановить юзерскрипты Greasemonkey из бэкапа браузера damix 15.01.2026 Если восстановить из бэкапа профиль Firefox после переустановки винды, то список юзерскриптов в Greasemonkey будет пустым. Но восстановить их можно так. Для этого понадобится консольная утилита. . .	Сукцессия микоризы: основная теория в виде двух уравнений. anaschu 11.01.2026 https:/ / rutube. ru/ video/ 7a537f578d808e67a3c6fd818a44a5c4/

@DonFerro 3 / 2 / 1 Регистрация: 10.07.2022 Сообщений: 11
	04.08.2022, 07:10 [ТС]
	Я пишу тесты для pytest, а не для unittest. 0

@Welemir1 Автоматизируй это! 7390 / 4817 / 1246 Регистрация: 30.03.2015 Сообщений: 13,667 Записей в блоге: 29
	04.08.2022, 09:00
	DonFerro, а, соррян, сбило с толку импорт мока, тогда да ассерты в питесте обычные(с виду). Беру свои слова назад, ничего менять больше не нужно 0

@DonFerro 3 / 2 / 1 Регистрация: 10.07.2022 Сообщений: 11
	04.08.2022, 09:43 [ТС]
	Welemir1, я правильно понимаю, что в pytest нет собственной реализации mock и нужно пользоваться сторонней, например от unittest? 0

@Welemir1 Автоматизируй это! 7390 / 4817 / 1246 Регистрация: 30.03.2015 Сообщений: 13,667 Записей в блоге: 29
	04.08.2022, 10:31
	DonFerro, да, у них нет мока, но есть манкипатч, он тоже предоставляет возможности подмен, посмотри на него. 1

Проверка вызова пропатченной функции

Решение