Update test_tractogram.py

Author: robbisg

nibabel/streamlines/tests/test_tractogram.py

@@ -603,20 +603,15 @@ def test_tractogram_copy(self):
 
         # Check we copied the data and not simply created new references.
         assert tractogram is not DATA['tractogram']
-        assert tractogram.streamlines is \
-                not DATA['tractogram'].streamlines
-        assert tractogram.data_per_streamline is \
-                not DATA['tractogram'].data_per_streamline
-        assert tractogram.data_per_point is \
-                not DATA['tractogram'].data_per_point
+        assert tractogram.streamlines is not DATA['tractogram'].streamlines
+        assert tractogram.data_per_streamline is not DATA['tractogram'].data_per_streamline
+        assert tractogram.data_per_point is not DATA['tractogram'].data_per_point
 
         for key in tractogram.data_per_streamline:
-            assert tractogram.data_per_streamline[key] is \
-                not DATA['tractogram'].data_per_streamline[key]
+            assert tractogram.data_per_streamline[key] is not DATA['tractogram'].data_per_streamline[key]
 
         for key in tractogram.data_per_point:
-            assert tractogram.data_per_point[key] is \
-                not DATA['tractogram'].data_per_point[key]
+            assert tractogram.data_per_point[key] is not DATA['tractogram'].data_per_point[key]
 
         # Check the values of the data are the same.
         assert_tractogram_equal(tractogram, DATA['tractogram'])
@@ -818,8 +813,14 @@ def test_lazy_tractogram_creation(self):
         # function.
         with pytest.raises(TypeError):
             LazyTractogram(streamlines=streamlines)
+            
+        with pytest.raises(TypeError):
             LazyTractogram(data_per_point={"none": None})
+            
+        with pytest.raises(TypeError):    
             LazyTractogram(data_per_streamline=data_per_streamline)
+            
+         with pytest.raises(TypeError):   
             LazyTractogram(streamlines=DATA['streamlines'],
                            data_per_point=data_per_point)
 
@@ -833,8 +834,8 @@ def test_lazy_tractogram_creation(self):
                                     DATA['data_per_streamline_func'],
                                     DATA['data_per_point_func'])
 
-        assert is_lazy_dict(tractogram.data_per_streamline) is True
-        assert is_lazy_dict(tractogram.data_per_point) is True
+        assert is_lazy_dict(tractogram.data_per_streamline)
+        assert is_lazy_dict(tractogram.data_per_point)
 
         [t for t in tractogram]  # Force iteration through tractogram.
         assert len(tractogram) == len(DATA['streamlines'])
@@ -1010,15 +1011,12 @@ def test_lazy_tractogram_copy(self):
 
         # When copying LazyTractogram, the generator function yielding
         # streamlines should stay the same.
-        assert tractogram._streamlines \
-                    is DATA['lazy_tractogram']._streamlines
+        assert tractogram._streamlines is DATA['lazy_tractogram']._streamlines
 
         # Copying LazyTractogram, creates new internal LazyDict objects,
         # but generator functions contained in it should stay the same.
-        assert tractogram._data_per_streamline \
-                    is not DATA['lazy_tractogram']._data_per_streamline
-        assert tractogram._data_per_point \
-                    is not DATA['lazy_tractogram']._data_per_point
+        assert tractogram._data_per_streamline is not DATA['lazy_tractogram']._data_per_streamline
+        assert tractogram._data_per_point is not DATA['lazy_tractogram']._data_per_point
 
         for key in tractogram.data_per_streamline:
             data = tractogram.data_per_streamline.store[key]
@@ -1031,8 +1029,7 @@ def test_lazy_tractogram_copy(self):
             assert data is expected
 
         # The affine should be a copy.
-        assert tractogram._affine_to_apply \
-                    is not DATA['lazy_tractogram']._affine_to_apply
+        assert tractogram._affine_to_apply is not DATA['lazy_tractogram']._affine_to_apply
         assert_array_equal(tractogram._affine_to_apply,
                            DATA['lazy_tractogram']._affine_to_apply)