From ce98a9219e060b729d4b0d2dc28bf4510649f0fd Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Mon, 17 Oct 2016 16:38:31 +1300 Subject: test & examples: zap six --- test/netlib/test_version_check.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'test/netlib/test_version_check.py') diff --git a/test/netlib/test_version_check.py b/test/netlib/test_version_check.py index fa6b19e5..01820dad 100644 --- a/test/netlib/test_version_check.py +++ b/test/netlib/test_version_check.py @@ -1,11 +1,11 @@ -from io import StringIO +import io import mock from netlib import version_check @mock.patch("sys.exit") def test_check_pyopenssl_version(sexit): - fp = StringIO() + fp = io.StringIO() version_check.check_pyopenssl_version(fp=fp) assert not fp.getvalue() assert not sexit.called @@ -19,7 +19,7 @@ def test_check_pyopenssl_version(sexit): @mock.patch("OpenSSL.__version__") def test_unparseable_pyopenssl_version(version, sexit): version.split.return_value = ["foo", "bar"] - fp = StringIO() + fp = io.StringIO() version_check.check_pyopenssl_version(fp=fp) assert "Cannot parse" in fp.getvalue() assert not sexit.called -- cgit v1.2.3