ImportError: Żaden moduł o nazwie mock

Więc próbuję użyć unittest.mock, aby podrobić niektóre z moich metod w moich testach jednostkowych. Ja robię:

from unittest.mock import MagicMock
f = open("data/static/mock_ffprobe_response")
subprocess.check_output = MagicMock(return_value=f.read())
f.close()

Ale dostaję:

ImportError: No module named mock

Próbowałem:

pip install mock

Nadal nie działa.

unittestto wbudowany moduł; mockjest biblioteką zewnętrzną (w każdym razie w wersji beta w wersji starszej niż 3.3). Po zainstalowaniu mockprzez pip install, importujesz go nie używając

from unittest.mock import MagicMock

ale

from mock import MagicMock

Edycja: mockzostała zawarta w unittestmodule (od Python3.3) i może być importowana przez import unittest.mock.

W przypadku Pythona 2.7:

Zainstaluj makietę:

pip install mock

Następnie w kodzie testowym użyj tego importu:

from mock import patch, MagicMock

Jeśli chcesz obsługiwać oba, Python 2 i Python 3, możesz również użyć następujących:

import sys
if sys.version_info >= (3, 3):
    from unittest.mock import MagicMock
else:
    from mock import MagicMock

lub, jeśli nie chcesz importować sys

try:
    from unittest.mock import MagicMock
except ImportError:
    from mock import MagicMock

Z jakiegoś powodu import unittest.mockmi nie wyszło, musiałem to zrobić

from unittest import mock

Podczas uruchamiania napotkałem problemy podczas importowania pytest. Okazało się, że moja pytestbyła to wersja 2.7 python, podczas gdy moja virtualenvbyła to 3.6 python. Naprawiłem to, wykonując pip installfrom my, virtualenva następnie używając pytestfrom env/bin/pytest.

Jeśli chodzi o twoją wersję, sugerowałbym udanie się na http://hg.python.org/cpython/file/default/Lib/unittest/mock.py i wykorzystanie tego źródła na swoją korzyść.

Zmiana

import mock

do

from unittest import mock

Dzieje się tak, ponieważ unittest.mock jest to wbudowany moduł zaczynający się od Pythona 3.3, ale istnieje backport w PyPI dla starszych wersji, który można zainstalować jako, pip install mocka następnie zaimportować jako import mock.

Jeśli używasz Pythona 3.3 lub nowszego, nie musisz niczego instalować za pomocą pip, wystarczy zaimportować go jako plik from unittest import mock.