Fix definition of write

sebastinas · sebastinas · commit 29d6f87f9688 · 2023-01-15T17:52:17.000+01:00
diff --git a/bpython/test/test_interpreter.py b/bpython/test/test_interpreter.py
@@ -12,9 +12,7 @@ class Interpreter(interpreter.Interp):
     def __init__(self):
         super().__init__()
         self.a = []
-
-    def write(self, data):
-        self.a.append(data)
+        self.write = self.a.append
 
 
 class TestInterpreter(unittest.TestCase):
