diff --git a/src/roiextractors/extractors/schnitzerextractor/extractsegmentationextractor.py b/src/roiextractors/extractors/schnitzerextractor/extractsegmentationextractor.py index b4bfd6a0..8376f76e 100644 --- a/src/roiextractors/extractors/schnitzerextractor/extractsegmentationextractor.py +++ b/src/roiextractors/extractors/schnitzerextractor/extractsegmentationextractor.py @@ -240,7 +240,8 @@ def get_images_dict(self): images_dict: dict dictionary with key, values representing different types of Images """ - images_dict = dict( + images_dict = super().get_images_dict() + images_dict.update( summary_image=self._info_struct["summary_image"][:], f_per_pixel=self._info_struct["F_per_pixel"][:], max_image=self._info_struct["max_image"][:], diff --git a/tests/test_extractsegmentationextractor.py b/tests/test_extractsegmentationextractor.py index 0a053460..06497ddb 100644 --- a/tests/test_extractsegmentationextractor.py +++ b/tests/test_extractsegmentationextractor.py @@ -199,7 +199,10 @@ def test_extractor_get_images_dict(self): )[:] images_dict = self.extractor.get_images_dict() - self.assertEqual(len(images_dict), 3) + self.assertEqual(len(images_dict), 5) + + self.assertEqual(images_dict["correlation"], None) + self.assertEqual(images_dict["mean"], None) self.assertEqual(images_dict["summary_image"].shape, summary_image.shape) self.assertEqual(images_dict["max_image"].shape, max_image.shape)