diff --git a/src/wormhole_transit_relay/test/test_rlimits.py b/src/wormhole_transit_relay/test/test_rlimits.py index 1354c40..3ee23a9 100644 --- a/src/wormhole_transit_relay/test/test_rlimits.py +++ b/src/wormhole_transit_relay/test/test_rlimits.py @@ -1,8 +1,5 @@ from __future__ import print_function, unicode_literals -try: - from unittest import mock -except ImportError: - import mock +from unittest import mock from twisted.trial import unittest from ..increase_rlimits import increase_rlimits diff --git a/src/wormhole_transit_relay/test/test_service.py b/src/wormhole_transit_relay/test/test_service.py index 003de32..f72765c 100644 --- a/src/wormhole_transit_relay/test/test_service.py +++ b/src/wormhole_transit_relay/test/test_service.py @@ -1,9 +1,6 @@ from __future__ import unicode_literals, print_function from twisted.trial import unittest -try: - from unittest import mock -except ImportError: - import mock +from unittest import mock from twisted.application.service import MultiService from .. import server_tap diff --git a/src/wormhole_transit_relay/test/test_stats.py b/src/wormhole_transit_relay/test/test_stats.py index 1f114b1..43b912f 100644 --- a/src/wormhole_transit_relay/test/test_stats.py +++ b/src/wormhole_transit_relay/test/test_stats.py @@ -1,9 +1,6 @@ from __future__ import print_function, unicode_literals import os, io, json, sqlite3 -try: - from unittest import mock -except ImportError: - import mock +from unittest import mock from twisted.trial import unittest from ..transit_server import Transit from .. import database