2 files changed, 8 insertions, 4 deletions
diff --git a/numpy/distutils/conv_template.py b/numpy/distutils/conv_template.py
index a42611051..4a8746236 100644
--- a/numpy/distutils/conv_template.py
+++ b/numpy/distutils/conv_template.py
@@ -315,8 +315,7 @@ def unique_key(adict):
     return newkey
 
 
-if __name__ == "__main__":
-
+def main():
     try:
         file = sys.argv[1]
     except IndexError:
@@ -335,3 +334,6 @@ if __name__ == "__main__":
         e = get_exception()
         raise ValueError("In %s loop at %s" % (file, e))
     outfile.write(writestr)
+
+if __name__ == "__main__":
+    main()
diff --git a/numpy/distutils/from_template.py b/numpy/distutils/from_template.py
index b19c7cc0b..8f587eab9 100644
--- a/numpy/distutils/from_template.py
+++ b/numpy/distutils/from_template.py
@@ -238,8 +238,7 @@ _special_names = find_repl_patterns('''
 <ctypereal=float,double,\\0,\\1>
 ''')
 
-if __name__ == "__main__":
-
+def main():
     try:
         file = sys.argv[1]
     except IndexError:
@@ -254,3 +253,6 @@ if __name__ == "__main__":
     allstr = fid.read()
     writestr = process_str(allstr)
     outfile.write(writestr)
+
+if __name__ == "__main__":
+    main()