Update Python snippet

Planeshifter · Planeshifter · commit 5e737ffedd89 · 2016-12-07T23:23:35.000-05:00
diff --git a/tools/snippets/test/fixtures/python/runner.py b/tools/snippets/test/fixtures/python/runner.py
@@ -1,48 +1,52 @@
 #!/usr/bin/env python
+"""Generate fixtures."""
 
+import os
+import json
+import math as m
 import numpy as np
 from scipy import special
-import math as m
-import json
-import os
 
-"""gen( x, name )
 
-Generate fixture data and write to file.
+# Get the file path:
+FILE = os.path.realpath( __file__ )
+
+# Extract the directory in which this file resides:
+DIR = os.path.dirname( file )
 
-# Arguments
+def gen( x, name ):
+    """Generate fixture data and write to file.
 
-* `x`: domain
-* `name::str`: output filename
+    # Arguments
 
-# Examples
+    * `x`: domain
+    * `name::str`: output filename
 
-``` python
-python> x = linspace( -1000, 1000, 2001 );
-python> gen( x, \"./data.json\" );
-```
-"""
-def gen( x, name ):
+    # Examples
+
+    ``` python
+    python> x = linspace( -1000, 1000, 2001 )
+    python> gen( x, \"./data.json\" )
+    ```
+    """
     # TODO: generate fixtures
 
     # Store data to be written to file as a dictionary:
     data = {
         "x": x.tolist(),
         "expected": y.tolist()
-    };
+    }
 
     # Based on the script directory, create an output filepath:
     filepath = os.path.join( dir, name )
 
     with open( filepath, 'w' ) as outfile:
         json.dump( data, outfile )
 
-# Get the file path:
-file = os.path.realpath( __file__ )
+def main():
+    """Generate fixture data."""
+    gen( x, "TODO" )
 
-# Extract the directory in which this file resides:
-dir = os.path.dirname( file )
 
-# Generate fixture data:
-# TODO: generate input data (`x`)
-gen( x, "TODO" );
+if __name__=="__main__":
+    main()
