improving solution and updating tests for related_files function

Author: Jaime Arias

nipype/utils/filemanip.py

@@ -30,7 +30,8 @@
 
 
 related_filetype_sets = [
-    ('.hdr', '.img', '.mat', '.nii'),
+    ('.hdr', '.img', '.mat'),
+    ('.nii', '.mat'),
     ('.BRIK', '.HEAD'),
 ]
 

nipype/utils/tests/test_filemanip.py

@@ -324,7 +324,7 @@ def test_json():
         ('/path/test.hdr',  3, ['/path/test.hdr', '/path/test.img', '/path/test.mat']),
         ('/path/test.BRIK', 2, ['/path/test.BRIK', '/path/test.HEAD']),
         ('/path/test.HEAD', 2, ['/path/test.BRIK', '/path/test.HEAD']),
-        ('/path/foo.nii',   1, [])
+        ('/path/foo.nii',   2, ['/path/foo.nii', '/path/foo.mat'])
         ])
 def test_related_files(file, length, expected_files):
     related_files = get_related_files(file)