PEP8

Author: mwaskom

nipype/interfaces/tests/test_io.py

@@ -11,11 +11,12 @@
 import nipype.interfaces.io as nio
 from nipype.interfaces.base import Undefined
 
+
 def test_datagrabber():
  dg = nio.DataGrabber()
  yield assert_equal, dg.inputs.template, Undefined
  yield assert_equal, dg.inputs.base_directory, Undefined
- yield assert_equal, dg.inputs.template_args,{'outfiles': []}
+ yield assert_equal, dg.inputs.template_args, {'outfiles': []}
 
 
 def test_selectfiles():
@@ -56,7 +57,7 @@ def test_datasink():
  yield assert_equal, ds.inputs.base_directory, Undefined
  yield assert_equal, ds.inputs.strip_dir, Undefined
  yield assert_equal, ds.inputs._outputs, {}
- ds = nio.DataSink(base_directory = 'foo')
+ ds = nio.DataSink(base_directory='foo')
  yield assert_equal, ds.inputs.base_directory, 'foo'
  ds = nio.DataSink(infields=['test'])
  yield assert_true, 'test' in ds.inputs.copyable_trait_names()
@@ -72,52 +73,58 @@ def test_datasink_substitutions():
  open(f, 'w')
  ds = nio.DataSink(
  parametrization=False,
- base_directory = outdir,
- substitutions = [('ababab', 'ABABAB')],
+ base_directory=outdir,
+ substitutions=[('ababab', 'ABABAB')],
  # end archoring ($) is used to assure operation on the filename
  # instead of possible temporary directories names matches
  # Patterns should be more comprehendable in the real-world usage
  # cases since paths would be quite more sensible
- regexp_substitutions = [(r'xABABAB(\w*)\.n$', r'a-\1-b.n'),
-  ('(.*%s)[-a]([^%s]*)$' % ((os.path.sep,)*2),
-  r'\1!\2')] )
+ regexp_substitutions=[(r'xABABAB(\w*)\.n$', r'a-\1-b.n'),
+ ('(.*%s)[-a]([^%s]*)$' % ((os.path.sep,) * 2),
+ r'\1!\2')])
  setattr(ds.inputs, '@outdir', files)
  ds.run()
  yield assert_equal, \
  sorted([os.path.basename(x) for
  x in glob.glob(os.path.join(outdir, '*'))]), \
- ['!-yz-b.n', 'ABABAB.n'] # so we got re used 2nd and both patterns
+ ['!-yz-b.n', 'ABABAB.n']  # so we got re used 2nd and both patterns
  shutil.rmtree(indir)
  shutil.rmtree(outdir)
 
+
 def _temp_analyze_files():
  """Generate temporary analyze file pair."""
- fd, orig_img = mkstemp(suffix = '.img', dir=mkdtemp())
+ fd, orig_img = mkstemp(suffix='.img', dir=mkdtemp())
  orig_hdr = orig_img[:-4] + '.hdr'
  fp = file(orig_hdr, 'w+')
  fp.close()
  return orig_img, orig_hdr
 
+
 def test_datasink_copydir():
  orig_img, orig_hdr = _temp_analyze_files()
  outdir = mkdtemp()
  pth, fname = os.path.split(orig_img)
- ds = nio.DataSink(base_directory = outdir, parameterization=False)
- setattr(ds.inputs,'@outdir',pth)
+ ds = nio.DataSink(base_directory=outdir, parameterization=False)
+ setattr(ds.inputs, '@outdir', pth)
  ds.run()
- file_exists = lambda: os.path.exists(os.path.join(outdir, pth.split(os.path.sep)[-1], fname))
+ sep = os.path.sep
+ file_exists = lambda: os.path.exists(os.path.join(outdir,
+ pth.split(sep)[-1],
+ fname))
  yield assert_true, file_exists()
  shutil.rmtree(pth)
 
  orig_img, orig_hdr = _temp_analyze_files()
  pth, fname = os.path.split(orig_img)
  ds.inputs.remove_dest_dir = True
- setattr(ds.inputs,'outdir',pth)
+ setattr(ds.inputs, 'outdir', pth)
  ds.run()
  yield assert_false, file_exists()
  shutil.rmtree(outdir)
  shutil.rmtree(pth)
 
+
 def test_freesurfersource():
  fss = nio.FreeSurferSource()
  yield assert_equal, fss.inputs.hemi, 'both'