diff --git a/tests/test_driver.py b/tests/test_driver.py index 4a5cd77..565b395 100644 --- a/tests/test_driver.py +++ b/tests/test_driver.py @@ -6,12 +6,7 @@ import uuid import errno import random from tempfile import NamedTemporaryFile as ntf - -try: - from cStringIO import StringIO -except ImportError: - from io import StringIO - +from io import StringIO from argparse import Namespace from tldptesttools import TestInventoryBase, TestToolsFilesystem diff --git a/tests/test_sources.py b/tests/test_sources.py index 4bb64e9..b1ac230 100644 --- a/tests/test_sources.py +++ b/tests/test_sources.py @@ -6,11 +6,7 @@ import errno import random import unittest from argparse import Namespace - -try: - from cStringIO import StringIO -except ImportError: - from io import StringIO +from io import StringIO from tldptesttools import TestToolsFilesystem