Fix tests

thomasballinger · thomasballinger · commit 4cb4e77a9230 · 2015-12-17T09:44:44.000-05:00
diff --git a/bpython/test/test_interpreter.py b/bpython/test/test_interpreter.py
@@ -2,7 +2,6 @@
 
 from __future__ import unicode_literals
 
-import linecache
 import sys
 
 from curtsies.fmtfuncs import bold, green, magenta, cyan, red, plain
@@ -14,12 +13,6 @@
 pypy = 'PyPy' in sys.version
 
 
-def _last_console_filename():
-    """Returns the last 'filename' used for console input
-    (as will be displayed in a traceback)."""
-    return '<bpython-input-%s>' % (len(linecache.cache.bpython_history) - 1)
-
-
 class TestInterpreter(unittest.TestCase):
     def test_syntaxerror(self):
         i = interpreter.Interp()
@@ -33,13 +26,13 @@ def append_to_a(message):
 
         if pypy:
             expected = (
-                '  File ' + green('"%s"' % _last_console_filename()) +
+                '  File ' + green('"<input>"') +
                 ', line ' + bold(magenta('1')) + '\n    1.1.1.1\n      ^\n' +
                 bold(red('SyntaxError')) + ': ' + cyan('invalid syntax') +
                 '\n')
         else:
             expected = (
-                '  File ' + green('"%s"' % _last_console_filename()) +
+                '  File ' + green('"<input>"') +
                 ', line ' + bold(magenta('1')) + '\n    1.1.1.1\n        ^\n' +
                 bold(red('SyntaxError')) + ': ' + cyan('invalid syntax') +
                 '\n')
@@ -62,7 +55,7 @@ def f():
         def g():
             return f()
 
-        i.runsource('g()', encode=False)
+        i.runsource('g()')
 
         if pypy:
             global_not_found = "global name 'g' is not defined"
@@ -71,7 +64,7 @@ def g():
 
         expected = (
             'Traceback (most recent call last):\n  File ' +
-            green('"%s"' % _last_console_filename()) + ', line ' +
+            green('"<input>"') + ', line ' +
             bold(magenta('1')) + ', in ' + cyan('<module>') + '\n    g()\n' +
             bold(red('NameError')) + ': ' + cyan(global_not_found) + '\n')
 
