added test for get_available_planes

catalystneuro · Sep 29, 2023 · dcaa04a · dcaa04a
1 parent ff9890d
commit dcaa04a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_scanimagetiffimagingextractor.py b/tests/test_scanimagetiffimagingextractor.py
@@ -19,6 +19,7 @@ def expected_properties():
         num_planes=2,
         frames_per_slice=2,
         channel_names=["Channel 1", "Channel 4"],
+        plane_names=["0", "1"],
         image_size=(528, 256),
         num_frames=6,
         dtype="int16",
@@ -166,6 +167,12 @@ def test_get_num_channels(scan_image_tiff_single_plane_imaging_extractor, expect
     assert num_channels == expected_properties["num_channels"]
 
 
+def test_get_available_planes(scan_image_tiff_single_plane_imaging_extractor, expected_properties):
+    file_path = str(scan_image_tiff_single_plane_imaging_extractor.file_path)
+    plane_names = ScanImageTiffSinglePlaneImagingExtractor.get_available_planes(file_path)
+    assert plane_names == expected_properties["plane_names"]
+
+
 def test_get_available_channels(scan_image_tiff_single_plane_imaging_extractor, expected_properties):
     file_path = str(scan_image_tiff_single_plane_imaging_extractor.file_path)
     channel_names = ScanImageTiffSinglePlaneImagingExtractor.get_available_channels(file_path)