Fix tests

Author: rly

tests/unit/test_ophys.py

@@ -1,6 +1,7 @@
 import warnings
 
 import numpy as np
+from hdmf.common import VectorData
 
 from pynwb.base import TimeSeries, ProcessingModule
 from pynwb.device import Device
@@ -142,7 +143,7 @@ def test_manifold_deprecated(self):
                 indicator='indicator',
                 location='location',
                 manifold=(1, 1, (2, 2, 2)))
-        
+
         # create object with deprecated argument
         with self.assertRaisesWith(ValueError, msg):
             ImagingPlane(**kwargs)
@@ -227,7 +228,7 @@ def test_init_description_optional(self):
                     'indicator',
                     'location',
                 )
-    
+
     def test_init_missing_required_args(self):
         """Check that ImagingPlane raises an error if required args are missing."""
         oc, device = self.set_up_dependencies()
@@ -527,7 +528,7 @@ def test_init(self):
         self.assertEqual(iS.name, 'test_iS')
         self.assertEqual(iS.plane_segmentations[ps.name], ps)
         self.assertEqual(iS[ps.name], iS.plane_segmentations[ps.name])
-    
+
     def test_add_segementation(self):
         ps = create_plane_segmentation()
         iS = ImageSegmentation(name='test_iS')
@@ -596,6 +597,7 @@ def test_init_no_name(self):
 
     def test_init_missing_roi_col_with_ids(self):
         """If no roi column is provided and ids were provided, an error should be raised"""
+        iSS, ip = self.set_up_dependencies()
         msg = "Must provide at least one of 'image_mask', 'pixel_mask', or 'voxel_mask' columns"
         with self.assertRaises(ValueError, msg=msg):
             PlaneSegmentation(
@@ -608,6 +610,7 @@ def test_init_missing_roi_col_with_ids(self):
 
     def test_init_missing_roi_col_with_columns(self):
         """If no roi column is provided and other non-empty columns are provided, an error should be raised"""
+        iSS, ip = self.set_up_dependencies()
         msg = "Must provide at least one of 'image_mask', 'pixel_mask', or 'voxel_mask' columns"
         with self.assertRaises(ValueError, msg=msg):
             PlaneSegmentation(
@@ -621,14 +624,15 @@ def test_init_missing_roi_col_with_columns(self):
 
     def test_init_missing_roi_col_with_empty_columns(self):
         """If no roi column is provided and other non-empty columns are provided, no error should be raised"""
+        iSS, ip = self.set_up_dependencies()
         pS = PlaneSegmentation(
             description='description',
             imaging_plane=ip,
             name='test_name',
             reference_images=iSS,
             columns=[VectorData(name="custom_col", description="custom col")],
         )
-        self.assertEqual(len(ps), 0)
+        self.assertEqual(len(pS), 0)
 
     def test_add_pixel_mask(self):
         pix_mask = [[1, 2, 1.0], [3, 4, 1.0], [5, 6, 1.0],