forked from pool/python-python-multipart
18 lines
477 B
Diff
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 *
|
||
|
|
|
||
|
|
|