fix test for early 2.7

thomasballinger · thomasballinger · commit ebeffd714c04 · 2016-07-28T22:51:21.000-04:00
diff --git a/bpython/test/test_args.py b/bpython/test/test_args.py
@@ -1,5 +1,6 @@
 # encoding: utf-8
 
+import re
 import subprocess
 import sys
 import tempfile
@@ -66,7 +67,11 @@ def test_exec_nonascii_file_linenums(self):
                 universal_newlines=True)
             (_, stderr) = p.communicate()
 
-            self.assertIn('line 3', stderr)
+            self.assertIn('line 3', clean_colors(stderr))
+
+
+def clean_colors(s):
+    return re.sub(r'\x1b[^m]*m', '', s)
 
 
 class TestParse(TestCase):
