From e34bba44ef53228aeba613ac6e928b2c315111cf Mon Sep 17 00:00:00 2001 From: Kevin Van Brunt Date: Wed, 26 Jun 2019 02:33:18 -0400 Subject: Moved code related to ANSI escape codes to new file called ansi.py --- cmd2/transcript.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'cmd2/transcript.py') 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) -- cgit v1.2.1