Protect tests/writetofile.py from test discovery

IPython/frontend/terminal/console/tests/writetofile.py

@@ -1,32 +1,33
 #-----------------------------------------------------------------------------
 # Copyright (C) 2012 The IPython Development Team
 #
 # Distributed under the terms of the BSD License. The full license is in
 # the file COPYING, distributed as part of this software.
 #-----------------------------------------------------------------------------
 
 """
 Copy data from input file to output file for testing.
 
 Command line usage:
 
     python writetofile.py INPUT OUTPUT
 
 Binary data from INPUT file is copied to OUTPUT file.
 If INPUT is '-', stdin is used.
 
 """
 
-import sys
-from IPython.utils.py3compat import PY3
-(inpath, outpath) = sys.argv[1:]
-
-if inpath == '-':
-    if PY3:
-        infile = sys.stdin.buffer
+if __name__ == '__main__':
+    import sys
+    from IPython.utils.py3compat import PY3
+    (inpath, outpath) = sys.argv[1:]
+
+    if inpath == '-':
+        if PY3:
+            infile = sys.stdin.buffer
+        else:
+            infile = sys.stdin
     else:
-        infile = sys.stdin
-else:
-    infile = open(inpath, 'rb')
+        infile = open(inpath, 'rb')
 
-open(outpath, 'w+b').write(infile.read())
+    open(outpath, 'w+b').write(infile.read())