TEST: Purge final get_data() calls from some tests

Author: effigies

nipype/interfaces/freesurfer/tests/test_model.py

@@ -32,12 +32,12 @@ def test_concatenate(tmpdir):
     # Test default behavior
     res = model.Concatenate(in_files=[in1, in2]).run()
     assert res.outputs.concatenated_file == tmpdir.join("concat_output.nii.gz").strpath
-    assert np.allclose(nb.load("concat_output.nii.gz").get_data(), out_data)
+    assert np.allclose(nb.load("concat_output.nii.gz").get_fdata(), out_data)
 
     # Test specified concatenated_file
     res = model.Concatenate(in_files=[in1, in2], concatenated_file=out).run()
     assert res.outputs.concatenated_file == tmpdir.join(out).strpath
-    assert np.allclose(nb.load(out, mmap=NUMPY_MMAP).get_data(), out_data)
+    assert np.allclose(nb.load(out, mmap=NUMPY_MMAP).get_fdata(), out_data)
 
     # Test in workflow
     wf = pe.Workflow("test_concatenate", base_dir=tmpdir.strpath)
@@ -47,12 +47,12 @@ def test_concatenate(tmpdir):
     wf.add_nodes([concat])
     wf.run()
     assert np.allclose(
-        nb.load(tmpdir.join("test_concatenate", "concat", out).strpath).get_data(),
+        nb.load(tmpdir.join("test_concatenate", "concat", out).strpath).get_fdata(),
         out_data,
     )
 
     # Test a simple statistic
     res = model.Concatenate(
         in_files=[in1, in2], concatenated_file=out, stats="mean"
     ).run()
-    assert np.allclose(nb.load(out, mmap=NUMPY_MMAP).get_data(), mean_data)
+    assert np.allclose(nb.load(out, mmap=NUMPY_MMAP).get_fdata(), mean_data)

nipype/interfaces/tests/test_image.py

@@ -50,15 +50,17 @@ def test_reorientation_backport():
 
         # Reorientation changes affine and data array
         assert not np.allclose(img.affine, reoriented_a.affine)
-        assert not (flips_only and np.allclose(img.get_data(), reoriented_a.get_data()))
+        assert not (
+            flips_only and np.allclose(img.get_fdata(), reoriented_a.get_fdata())
+        )
         # Dimension info changes iff axes are reordered
         assert flips_only == np.array_equal(
             img.header.get_dim_info(), reoriented_a.header.get_dim_info()
         )
 
         # Both approaches produce equivalent images
         assert np.allclose(reoriented_a.affine, reoriented_b.affine)
-        assert np.array_equal(reoriented_a.get_data(), reoriented_b.get_data())
+        assert np.array_equal(reoriented_a.get_fdata(), reoriented_b.get_fdata())
         assert np.array_equal(
             reoriented_a.header.get_dim_info(), reoriented_b.header.get_dim_info()
         )