Address review comments

Author: mwcraig

src/astro_image_display_api/api_test.py

@@ -801,20 +801,24 @@ def test_save_overwrite(self, tmp_path):
     def test_get_image_labels(self, data):
         # the test viewer begins with a default empty image
         assert len(self.image.get_image_labels()) == 1
-        assert self.image.get_image_labels()[0] == None
+        assert self.image.get_image_labels()[0] is None
         assert isinstance(self.image.get_image_labels(), tuple)
 
         self.image.load_image(data, image_label="test")
         assert len(self.image.get_image_labels()) == 2
-        assert self.image.get_image_labels()[-1] == 'test'
+        assert self.image.get_image_labels()[-1] == "test"
 
     def test_get_image(self, data):
         self.image.load_image(data, image_label="test")
 
         # currently the type is not specified in the API
         assert self.image.get_image() is not None
         assert self.image.get_image(image_label="test") is not None
-        
+
+        retrieved_image = self.image.get_image(image_label="test")
+
+        self.image.load_image(retrieved_image, image_label="another test")
+        assert self.image.get_image(image_label="another test") is not None
+
         with pytest.raises(ValueError, match="[Ii]mage label.*not found"):
             self.image.get_image(image_label="not a valid label")
-

src/astro_image_display_api/interface_definition.py

@@ -60,12 +60,6 @@ def get_image(
             The label of the image to set the cuts for. If not given and there is
             only one image loaded, that image is returned.
 
-        Raises
-        ------
-        ValueError
-            If the `image_label` is not provided when there are multiple images loaded,
-            or if the `image_label` does not correspond to a loaded image.
-
         Returns
         -------
         image_data : Any
@@ -74,13 +68,20 @@ def get_image(
             with `astropy.nddata.NDData` is preferred, but not required. It is expected
             that the returned data can be re-loaded into the viewer using
             `load_image`, however.
+
+        Raises
+        ------
+        ValueError
+            If the `image_label` is not provided when there are multiple images loaded,
+            or if the `image_label` does not correspond to a loaded image.
+
         """
         raise NotImplementedError
-        
+
     @abstractmethod
     def get_image_labels(
         self,
-    ) -> tuple[str]: 
+    ) -> tuple[str]:
         """
         Get the labels of the loaded images.