diff options
Diffstat (limited to 'tests/test_cmd2.py')
| -rw-r--r-- | tests/test_cmd2.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/tests/test_cmd2.py b/tests/test_cmd2.py index 8435ac1a..0deb9a0b 100644 --- a/tests/test_cmd2.py +++ b/tests/test_cmd2.py @@ -1,3 +1,4 @@ +# coding=utf-8 # # Cmd2 unit/functional testing # @@ -6,14 +7,10 @@ import mock import pytest - from conftest import run_cmd, _normalize -import cmd2 +from six import StringIO -try: - from StringIO import StringIO -except ImportError: - from io import StringIO +import cmd2 def test_ver(): |