forked from pool/python-flaky
Matej Cepl
1752fcd3a7
- We don't need to break Python 2.7 OBS-URL: https://build.opensuse.org/request/show/855604 OBS-URL: https://build.opensuse.org/package/show/devel:languages:python/python-flaky?expand=0&rev=16
15 lines
404 B
Diff
15 lines
404 B
Diff
--- a/test/test_pytest/test_flaky_pytest_plugin.py
|
|
+++ b/test/test_pytest/test_flaky_pytest_plugin.py
|
|
@@ -2,7 +2,10 @@
|
|
|
|
from __future__ import unicode_literals
|
|
from io import StringIO
|
|
-from mock import Mock, patch
|
|
+try:
|
|
+ from unittest.mock import Mock, patch
|
|
+except ImportError:
|
|
+ from mock import Mock, patch
|
|
# pylint:disable=import-error
|
|
import pytest
|
|
from _pytest.runner import CallInfo
|