summaryrefslogtreecommitdiff
path: root/cmd2/transcript.py
diff options
context:
space:
mode:
authorTodd Leonhardt <todd.leonhardt@gmail.com>2021-01-31 22:29:57 -0500
committerTodd Leonhardt <todd.leonhardt@gmail.com>2021-01-31 22:29:57 -0500
commitf456b802754c3d1095b488d670bebba21018d823 (patch)
treebf7011f585a8a35ab4cc7ff98aaeebed4de87f84 /cmd2/transcript.py
parent918200c02d392c17862fff81bbf58820ed15c725 (diff)
downloadcmd2-git-f456b802754c3d1095b488d670bebba21018d823.tar.gz
Add black for automatic code format
Diffstat (limited to 'cmd2/transcript.py')
-rw-r--r--cmd2/transcript.py28
1 files changed, 15 insertions, 13 deletions
diff --git a/cmd2/transcript.py b/cmd2/transcript.py
index 0c65cb8a..f45ac73d 100644
--- a/cmd2/transcript.py
+++ b/cmd2/transcript.py
@@ -11,9 +11,7 @@ class is used in cmd2.py::run_transcript_tests()
"""
import re
import unittest
-from typing import (
- Tuple,
-)
+from typing import Tuple
from . import (
ansi,
@@ -30,6 +28,7 @@ class Cmd2TestCase(unittest.TestCase):
See example.py
"""
+
cmdapp = None
def setUp(self):
@@ -72,7 +71,7 @@ class Cmd2TestCase(unittest.TestCase):
finished = True
break
line_num += 1
- command = [line[len(self.cmdapp.visible_prompt):]]
+ command = [line[len(self.cmdapp.visible_prompt) :]]
try:
line = next(transcript)
except StopIteration:
@@ -80,12 +79,13 @@ class Cmd2TestCase(unittest.TestCase):
line_num += 1
# Read the entirety of a multi-line command
while line.startswith(self.cmdapp.continuation_prompt):
- command.append(line[len(self.cmdapp.continuation_prompt):])
+ command.append(line[len(self.cmdapp.continuation_prompt) :])
try:
line = next(transcript)
except StopIteration as exc:
- msg = 'Transcript broke off while reading command beginning at line {} with\n{}'.format(line_num,
- command[0])
+ msg = 'Transcript broke off while reading command beginning at line {} with\n{}'.format(
+ line_num, command[0]
+ )
raise StopIteration(msg) from exc
line_num += 1
command = ''.join(command)
@@ -96,7 +96,8 @@ class Cmd2TestCase(unittest.TestCase):
# Read the expected result from transcript
if ansi.strip_style(line).startswith(self.cmdapp.visible_prompt):
message = '\nFile {}, line {}\nCommand was:\n{}\nExpected: (nothing)\nGot:\n{}\n'.format(
- fname, line_num, command, result)
+ fname, line_num, command, result
+ )
self.assertTrue(not (result.strip()), message)
# If the command signaled the application to quit there should be no more commands
self.assertFalse(stop, stop_msg)
@@ -119,7 +120,8 @@ class Cmd2TestCase(unittest.TestCase):
expected = ''.join(expected)
expected = self._transform_transcript_expected(expected)
message = '\nFile {}, line {}\nCommand was:\n{}\nExpected:\n{}\nGot:\n{}\n'.format(
- fname, line_num, command, expected, result)
+ fname, line_num, command, expected, result
+ )
self.assertTrue(re.match(expected, result, re.MULTILINE | re.DOTALL), message)
def _transform_transcript_expected(self, s: str) -> str:
@@ -165,7 +167,7 @@ class Cmd2TestCase(unittest.TestCase):
else:
# No closing slash, we have to add the first slash,
# and the rest of the text
- regex += re.escape(s[start - 1:])
+ regex += re.escape(s[start - 1 :])
break
return regex
@@ -192,18 +194,18 @@ class Cmd2TestCase(unittest.TestCase):
break
else:
# check if the slash is preceeded by a backslash
- if s[pos - 1:pos] == '\\':
+ if s[pos - 1 : pos] == '\\':
# it is.
if in_regex:
# add everything up to the backslash as a
# regular expression
- regex += s[start:pos - 1]
+ regex += s[start : pos - 1]
# skip the backslash, and add the slash
regex += s[pos]
else:
# add everything up to the backslash as escaped
# plain text
- regex += re.escape(s[start:pos - 1])
+ regex += re.escape(s[start : pos - 1])
# and then add the slash as escaped
# plain text
regex += re.escape(s[pos])