Last test of LabelMap

nd2reader/artificial.py

@@ -78,7 +78,8 @@ class ArtificialND2(object):
             'lut_data': "CustomDataVar|LUTDataV1_0!",
             'grabber_settings': "CustomDataVar|GrabberCameraSettingsV1_0!",
             'custom_data': "CustomDataVar|CustomDataV2_0!",
-            'app_info': "CustomDataVar|AppInfo_V1_0!"
+            'app_info': "CustomDataVar|AppInfo_V1_0!",
+            'image_frame_0': "ImageDataSeq|0!"
         }
         data = {}
 

nd2reader/label_map.py

@@ -12,15 +12,6 @@ class LabelMap(object):
         self._data = raw_binary_data
         self._image_data = {}
 
-    def image_attributes(self):
-        """Get the location of the image attributes
-
-        Returns:
-            int: The location of the image attributes
-
-        """
-        return self._get_location(six.b("ImageAttributesLV!"))
-
     def _get_location(self, label):
         try:
             label_location = self._data.index(label) + len(label)

tests/test_label_map.py

@@ -8,6 +8,9 @@ class TestLabelMap(unittest.TestCase):
         self.raw_text, self.locations = ArtificialND2.create_label_map_bytes()
         self.label_map = LabelMap(self.raw_text)
 
+    def test_image_data_location(self):
+        self.assertEqual(self.locations['image_frame_0'][0], self.label_map.get_image_data_location(0))
+
     def test_image_text_info(self):
         self.assertEqual(self.locations['image_text_info'][0], self.label_map.image_text_info)