summaryrefslogtreecommitdiff
path: root/cmd2/transcript.py
diff options
context:
space:
mode:
authorKevin Van Brunt <kmvanbrunt@gmail.com>2019-06-26 02:33:18 -0400
committerKevin Van Brunt <kmvanbrunt@gmail.com>2019-06-26 02:33:18 -0400
commite34bba44ef53228aeba613ac6e928b2c315111cf (patch)
treea39a736c3a36426fe6c278105ce9f6ec5696d16c /cmd2/transcript.py
parente6f65e607c6ad2350dfce05f3d5080d0ffbe35f1 (diff)
downloadcmd2-git-e34bba44ef53228aeba613ac6e928b2c315111cf.tar.gz
Moved code related to ANSI escape codes to new file called ansi.py
Diffstat (limited to 'cmd2/transcript.py')
-rw-r--r--cmd2/transcript.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/cmd2/transcript.py b/cmd2/transcript.py
index 316592ce..25a79310 100644
--- a/cmd2/transcript.py
+++ b/cmd2/transcript.py
@@ -13,7 +13,7 @@ import re
import unittest
from typing import Tuple
-from . import utils
+from . import ansi, utils
class Cmd2TestCase(unittest.TestCase):
@@ -56,13 +56,13 @@ class Cmd2TestCase(unittest.TestCase):
def _test_transcript(self, fname: str, transcript):
line_num = 0
finished = False
- line = utils.strip_ansi(next(transcript))
+ line = ansi.strip_ansi(next(transcript))
line_num += 1
while not finished:
# Scroll forward to where actual commands begin
while not line.startswith(self.cmdapp.visible_prompt):
try:
- line = utils.strip_ansi(next(transcript))
+ line = ansi.strip_ansi(next(transcript))
except StopIteration:
finished = True
break
@@ -89,7 +89,7 @@ class Cmd2TestCase(unittest.TestCase):
result = self.cmdapp.stdout.read()
stop_msg = 'Command indicated application should quit, but more commands in transcript'
# Read the expected result from transcript
- if utils.strip_ansi(line).startswith(self.cmdapp.visible_prompt):
+ if ansi.strip_ansi(line).startswith(self.cmdapp.visible_prompt):
message = '\nFile {}, line {}\nCommand was:\n{}\nExpected: (nothing)\nGot:\n{}\n'.format(
fname, line_num, command, result)
self.assertTrue(not (result.strip()), message)
@@ -97,7 +97,7 @@ class Cmd2TestCase(unittest.TestCase):
self.assertFalse(stop, stop_msg)
continue
expected = []
- while not utils.strip_ansi(line).startswith(self.cmdapp.visible_prompt):
+ while not ansi.strip_ansi(line).startswith(self.cmdapp.visible_prompt):
expected.append(line)
try:
line = next(transcript)