Fix tests

aleborba · Oct 30, 2018 · 9f27ffc · 9f27ffc
1 parent ced5353
commit 9f27ffc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/correios_api_test.py b/tests/correios_api_test.py
@@ -1,6 +1,7 @@
 import unittest
 
-from mockito import when, Mock
+from mock import Mock
+from mockito import when
 
 from packtrack import Correios
 

diff --git a/tests/correios_test.py b/tests/correios_test.py
@@ -1,5 +1,6 @@
 import unittest
 
+from mock import Mock
 from mockito import *
 
 from packtrack.correios import Encomenda, Status, EncomendaRepository