Small (subjective) clean-up in tests.

Trundle · Trundle · commit d7afe7350654 · 2014-10-17T21:32:24.000+02:00
diff --git a/bpython/test/test_args.py b/bpython/test/test_args.py
@@ -1,19 +1,22 @@
 import unittest
 import subprocess
-import tempfile
 import sys
-
-
+import tempfile
+from textwrap import dedent
 
 
 class TestExecArgs(unittest.TestCase):
     def test_exec_dunder_file(self):
-        with tempfile.NamedTemporaryFile(delete=False) as f:
-            f.write(
-            "import sys; sys.stderr.write(__file__); sys.stderr.flush();".encode('ascii'))
+        with tempfile.NamedTemporaryFile(delete=False, mode="w") as f:
+            f.write(dedent("""\
+                import sys
+                sys.stderr.write(__file__)
+                sys.stderr.flush()"""))
             f.flush()
-            p = subprocess.Popen(['bpython-curtsies', f.name], stderr=subprocess.PIPE)
-
-            self.assertEquals(p.stderr.read().strip().decode('ascii'), f.name)
-
+            p = subprocess.Popen(
+                [sys.executable, "-m", "bpython.curtsies", f.name],
+                stderr=subprocess.PIPE,
+                universal_newlines=True)
+            (_, stderr) = p.communicate()
 
+            self.assertEquals(stderr.strip(), f.name)
