STY: pep8 clean

Author: effigies

nibabel/freesurfer/io.py

@@ -4,6 +4,7 @@
 import getpass
 import time
 
+
 def _fread3(fobj):
     """Read a 3-byte int from an open binary file object
 
@@ -93,6 +94,7 @@ def read_geometry(filepath):
     coords = coords.astype(np.float)  # XXX: due to mayavi bug on mac 32bits
     return coords, faces
 
+
 def write_geometry(filepath, coords, faces, create_stamp=None):
     """Write a triangular format Freesurfer surface mesh.
 
@@ -107,10 +109,11 @@ def write_geometry(filepath, coords, faces, create_stamp=None):
     create_stamp : str
         User/time stamp (default: "created by <user> on <ctime>")
     """
-    magic_bytes = np.array([255,255,254],dtype=np.uint8)
+    magic_bytes = np.array([255, 255, 254], dtype=np.uint8)
 
     if create_stamp is None:
-        create_stamp = "created by %s on %s" % (getpass.getuser(), time.ctime())
+        create_stamp = "created by %s on %s" % (getpass.getuser(),
+            time.ctime())
 
     with open(filepath, 'wb') as fobj:
         magic_bytes.tofile(fobj)
@@ -122,6 +125,7 @@ def write_geometry(filepath, coords, faces, create_stamp=None):
         coords.astype('>f4').reshape(-1).tofile(fobj)
         faces.astype('>i4').reshape(-1).tofile(fobj)
 
+
 def read_morph_data(filepath):
     """Read a Freesurfer morphometry data file.
 

nibabel/freesurfer/tests/test_io.py

@@ -10,7 +10,8 @@
 import numpy as np
 from numpy.testing import assert_equal
 
-from .. import read_geometry, read_morph_data, read_annot, read_label, write_geometry
+from .. import read_geometry, read_morph_data, read_annot, read_label, \
+                write_geometry
 
 
 have_freesurfer = True
@@ -45,12 +46,13 @@ def test_geometry():
     # with respect to read_geometry()
     with InTemporaryDirectory():
         surf_path = 'test'
-        create_stamp = "created by %s on %s" % (getpass.getuser(), time.ctime())
+        create_stamp = "created by %s on %s" % (getpass.getuser(),
+            time.ctime())
         write_geometry(surf_path, coords, faces, create_stamp)
 
         coords2, faces2 = read_geometry(surf_path)
 
-        with open(surf_path,'rb') as fobj:
+        with open(surf_path, 'rb') as fobj:
             magic = np.fromfile(fobj, ">u1", 3)
             read_create_stamp = fobj.readline().rstrip('\n')
 
@@ -65,6 +67,7 @@ def test_geometry():
     np.testing.assert_array_equal(coords_swapped, coords)
     np.testing.assert_array_equal(faces_swapped, faces)
 
+
 @freesurfer_test
 def test_morph_data():
     """Test IO of morphometry data file (eg. curvature)."""