summaryrefslogtreecommitdiff
path: root/tests/test_transcript.py
diff options
context:
space:
mode:
authorEric Lin <anselor@gmail.com>2018-04-15 23:24:03 -0400
committerEric Lin <anselor@gmail.com>2018-04-15 23:24:03 -0400
commitbfc190e64481c161929d69e358c0dccb32d67c21 (patch)
tree96b12312e91ec3497faf0adce547656ed1431387 /tests/test_transcript.py
parent8d4f8411bb5f6bbac8ec121f897ccece1b97006c (diff)
parent480c7f8bbcdf1179312f71f525e79102aa4551ae (diff)
downloadcmd2-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.py14
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', [