Always use packaged mock.

apache · Jul 13, 2016 · d9fc1ed · d9fc1ed
1 parent 01c6db4
commit d9fc1ed
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/libcloud/test/compute/test_libvirt_driver.py b/libcloud/test/compute/test_libvirt_driver.py
@@ -15,15 +15,15 @@
 
 import sys
 
-from libcloud.compute.drivers.libvirt_driver import LibvirtNodeDriver
-from libcloud.utils.py3 import unittest2_required
+import mock
 
+from libcloud.compute.drivers.libvirt_driver import LibvirtNodeDriver
 
 from libcloud.test import unittest
-if unittest2_required:
-    import mock
-else:
-    from unittest import mock
+
+__all__ = [
+    'LibvirtNodeDriverTestCase'
+]
 
 
 @mock.patch('libcloud.compute.drivers.libvirt_driver.libvirt', autospec=True)