diff options
| -rwxr-xr-x | tests/test_ansi.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_ansi.py b/tests/test_ansi.py index 2140013..3d73fe8 100755 --- a/tests/test_ansi.py +++ b/tests/test_ansi.py @@ -121,19 +121,19 @@ class ansiTestCase (PexpectTestCase.PexpectTestCase): def test_torturet (self): s = ANSI.ANSI (24,80) + with open('torturet.vt') as f: + sample_text = f.read() # This causes ANSI.py's DoLog to write in the cwd. Make sure we're in a # writeable directory. d = tempfile.mkdtemp() old_cwd = os.getcwd() os.chdir(d) try: - with open('torturet.vt') as f: - sample_text = f.read() + for c in sample_text: + s.process (c) finally: os.chdir(old_cwd) shutil.rmtree(d) - for c in sample_text: - s.process (c) assert s.pretty() == torture_target, 'processed: \n' + s.pretty() + '\nexpected:\n' + torture_target def test_tetris (self): |