python-python-multipart/python-python-multipart-no-mock.patch
2022-04-08 20:46:03 +00:00

18 lines
477 B
Diff

diff --git a/multipart/tests/test_multipart.py b/multipart/tests/test_multipart.py
index 5769905..7913cd2 100644
--- a/multipart/tests/test_multipart.py
+++ b/multipart/tests/test_multipart.py
@@ -16,7 +16,10 @@
from io import BytesIO
from six import binary_type, text_type
-from mock import MagicMock, Mock, patch
+try:
+ from unittest.mock import MagicMock, Mock, patch
+except ImportError:
+ from mock import MagicMock, Mock, patch
from ..multipart import *