diff options
author | Eric Lin <anselor@gmail.com> | 2018-04-15 23:24:03 -0400 |
---|---|---|
committer | Eric Lin <anselor@gmail.com> | 2018-04-15 23:24:03 -0400 |
commit | bfc190e64481c161929d69e358c0dccb32d67c21 (patch) | |
tree | 96b12312e91ec3497faf0adce547656ed1431387 /tests/test_transcript.py | |
parent | 8d4f8411bb5f6bbac8ec121f897ccece1b97006c (diff) | |
parent | 480c7f8bbcdf1179312f71f525e79102aa4551ae (diff) | |
download | cmd2-git-bfc190e64481c161929d69e358c0dccb32d67c21.tar.gz |
Merge remote-tracking branch 'origin/python3' into autocompleter
Diffstat (limited to 'tests/test_transcript.py')
-rw-r--r-- | tests/test_transcript.py | 14 |
1 files changed, 4 insertions, 10 deletions
diff --git a/tests/test_transcript.py b/tests/test_transcript.py index f7b4a8f2..a24f2fa5 100644 --- a/tests/test_transcript.py +++ b/tests/test_transcript.py @@ -11,9 +11,8 @@ import sys import re import random -import mock +from unittest import mock import pytest -import six import cmd2 from cmd2 import Cmd, Cmd2TestCase, set_posix_shlex, set_strip_quotes @@ -33,8 +32,7 @@ class CmdLineApp(Cmd): # Add stuff to settable and/or shortcuts before calling base class initializer self.settable['maxrepeats'] = 'Max number of `--repeat`s allowed' - # Need to use this older form of invoking super class constructor to support Python 2.x and Python 3.x - Cmd.__init__(self, *args, **kwargs) + super().__init__(*args, **kwargs) self.intro = 'This is an intro banner ...' # Configure how arguments are parsed for commands using decorators @@ -267,12 +265,8 @@ def test_transcript(request, capsys, filename, feedback_to_output): expected_start = ".\n----------------------------------------------------------------------\nRan 1 test in" expected_end = "s\n\nOK\n" out, err = capsys.readouterr() - if six.PY3: - assert err.startswith(expected_start) - assert err.endswith(expected_end) - else: - assert err == '' - assert out == '' + assert err.startswith(expected_start) + assert err.endswith(expected_end) @pytest.mark.parametrize('expected, transformed', [ |