diff options
Diffstat (limited to 'tests/test_selectors.py')
-rw-r--r-- | tests/test_selectors.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tests/test_selectors.py b/tests/test_selectors.py index 9392962..a766f9f 100644 --- a/tests/test_selectors.py +++ b/tests/test_selectors.py @@ -1,9 +1,10 @@ """Tests for selectors.py.""" import unittest -from unittest import mock -from asyncio import selectors +from trollius import selectors +from trollius import test_utils +from trollius.test_utils import mock class FakeSelector(selectors._BaseSelectorImpl): @@ -13,7 +14,7 @@ class FakeSelector(selectors._BaseSelectorImpl): raise NotImplementedError -class _SelectorMappingTests(unittest.TestCase): +class _SelectorMappingTests(test_utils.TestCase): def test_len(self): s = FakeSelector() @@ -60,7 +61,7 @@ class _SelectorMappingTests(unittest.TestCase): self.assertEqual(1, counter) -class BaseSelectorTests(unittest.TestCase): +class BaseSelectorTests(test_utils.TestCase): def test_fileobj_to_fd(self): self.assertEqual(10, selectors._fileobj_to_fd(10)) |