
import os
from tempfile import mkdtemp
from shutil import rmtree

from nipy.testing import *
from sneeze import find_pkg, run_nose

import_strings = ["from nipype.interfaces.afni import To3d",
                  "from nipype.interfaces import afni",
                  "import nipype.interfaces.afni"]

def test_from_namespace():
    dname = mkdtemp()
    fname = os.path.join(dname, 'test_afni.py')
    fp = open(fname, 'w')
    fp.write('from nipype.interfaces.afni import To3d')
    fp.close()
    cover_pkg, module = find_pkg(fname)
    cmd = run_nose(cover_pkg, fname, dry_run=True)
    cmdlst = cmd.split()
    cmd = ' '.join(cmdlst[:4]) # strip off temporary directory path
    #print cmd
    assert_equal(cmd, 'nosetests -sv --with-coverage --cover-package=nipype.interfaces.afni')
    if os.path.exists(dname):
        rmtree(dname)


import os
from tempfile import mkdtemp
from shutil import rmtree

from nipy.testing import *
from sneeze import find_pkg, run_nose

import_strings = ["from nipype.interfaces.afni import To3d, ThreeDRefit",
                  "from nipype.interfaces import afni",
                  "import nipype.interfaces.afni",
                  "from nipype.interfaces import afni as af"]

def test_imports():
    dname = mkdtemp()
    fname = os.path.join(dname, 'test_afni.py')
    for impt in import_strings:
        fp = open(fname, 'w')
        fp.write(impt)
        fp.close()
        cover_pkg, module = find_pkg(fname)
        cmd = run_nose(cover_pkg, fname, dry_run=True)
        cmdlst = cmd.split()
        cmd = ' '.join(cmdlst[:4]) # strip off temporary directory path
        yield assert_equal, cmd, \
        'nosetests -sv --with-coverage --cover-package=nipype.interfaces.afni'
    if os.path.exists(dname):
        rmtree(dname)

Add more tests for sneeze.
Add more tests for sneeze.
  
  import os
  from tempfile import mkdtemp
  from shutil import rmtree
  
  from nipy.testing import *
  from sneeze import find_pkg, run_nose
  
- import_strings = ["from nipype.interfaces.afni import To3d",
+ import_strings = ["from nipype.interfaces.afni import To3d, ThreeDRefit",
                    "from nipype.interfaces import afni",
-                   "import nipype.interfaces.afni"]
+                   "import nipype.interfaces.afni",
+                   "from nipype.interfaces import afni as af"]
  
- def test_from_namespace():
+ def test_imports():
      dname = mkdtemp()
      fname = os.path.join(dname, 'test_afni.py')
+     for impt in import_strings:
-     fp = open(fname, 'w')
+         fp = open(fname, 'w')
-     fp.write('from nipype.interfaces.afni import To3d')
+         fp.write(impt)
-     fp.close()
+         fp.close()
-     cover_pkg, module = find_pkg(fname)
+         cover_pkg, module = find_pkg(fname)
-     cmd = run_nose(cover_pkg, fname, dry_run=True)
+         cmd = run_nose(cover_pkg, fname, dry_run=True)
-     cmdlst = cmd.split()
+         cmdlst = cmd.split()
-     cmd = ' '.join(cmdlst[:4]) # strip off temporary directory path
+         cmd = ' '.join(cmdlst[:4]) # strip off temporary directory path
-     #print cmd
+         yield assert_equal, cmd, \
-     assert_equal(cmd, 'nosetests -sv --with-coverage --cover-package=nipype.interfaces.afni')
+         'nosetests -sv --with-coverage --cover-package=nipype.interfaces.afni'
      if os.path.exists(dname):
          rmtree(dname)
  
Add more tests for sneeze.
## Code Before:

import os
from tempfile import mkdtemp
from shutil import rmtree

from nipy.testing import *
from sneeze import find_pkg, run_nose

import_strings = ["from nipype.interfaces.afni import To3d",
                  "from nipype.interfaces import afni",
                  "import nipype.interfaces.afni"]

def test_from_namespace():
    dname = mkdtemp()
    fname = os.path.join(dname, 'test_afni.py')
    fp = open(fname, 'w')
    fp.write('from nipype.interfaces.afni import To3d')
    fp.close()
    cover_pkg, module = find_pkg(fname)
    cmd = run_nose(cover_pkg, fname, dry_run=True)
    cmdlst = cmd.split()
    cmd = ' '.join(cmdlst[:4]) # strip off temporary directory path
    #print cmd
    assert_equal(cmd, 'nosetests -sv --with-coverage --cover-package=nipype.interfaces.afni')
    if os.path.exists(dname):
        rmtree(dname)

## Instruction:
Add more tests for sneeze.
## Code After:

import os
from tempfile import mkdtemp
from shutil import rmtree

from nipy.testing import *
from sneeze import find_pkg, run_nose

import_strings = ["from nipype.interfaces.afni import To3d, ThreeDRefit",
                  "from nipype.interfaces import afni",
                  "import nipype.interfaces.afni",
                  "from nipype.interfaces import afni as af"]

def test_imports():
    dname = mkdtemp()
    fname = os.path.join(dname, 'test_afni.py')
    for impt in import_strings:
        fp = open(fname, 'w')
        fp.write(impt)
        fp.close()
        cover_pkg, module = find_pkg(fname)
        cmd = run_nose(cover_pkg, fname, dry_run=True)
        cmdlst = cmd.split()
        cmd = ' '.join(cmdlst[:4]) # strip off temporary directory path
        yield assert_equal, cmd, \
        'nosetests -sv --with-coverage --cover-package=nipype.interfaces.afni'
    if os.path.exists(dname):
        rmtree(dname)
