Testing:Expand Test coverage

rsciio/dm5/_api.py

@@ -463,9 +463,13 @@ def update_dimension(self, axis, length=None):
         """
         Update the dimension of the image for a given axis.
 
-        This is two places in the DM5 file???
-
-        Under Calibrations and under Dimension. I think that only the Calibrations should be updated.
+        Parameters
+        ----------
+        axis : int
+            The axis to update the dimension for (Starting from 0 in array order). This will be reversed to match DM's
+            axis order.
+        length : int, optional
+            The length of the axis.
 
         """
         axis = self.ndim - axis - 1
@@ -485,6 +489,12 @@ def _get_dimension(self, axis):
     def get_data(self, lazy=False):
         """
         Get the image data.
+
+        Parameters
+        ----------
+        lazy : bool, optional
+            Whether to return a dask array or a numpy
+
         """
         if lazy:
             return da.from_array(self.image_data["Data"])
@@ -494,6 +504,11 @@ def get_data(self, lazy=False):
     def update_data(self, data):
         """
         Update the image data.
+
+        Parameters
+        ----------
+        data : np.ndarray or da.Array
+            The new image data.
         """
         HyperspyWriter.overwrite_dataset(self.image_data, data, "Data")
         self.image_data.attrs.update(
@@ -522,7 +537,7 @@ def get_metadata(self):
         if "Microscope Info" in original_metadata:
             metadata["Acquisition_instrument"] = {}
             metadata["Acquisition_instrument"]["TEM"] = {}
-            metadata["Acquisition_instrument"]["TEM"]["beam_energy "] = (
+            metadata["Acquisition_instrument"]["TEM"]["beam_energy"] = (
                 original_metadata["Microscope Info"].get("Voltage", 0) / 1000
             )
             metadata["Acquisition_instrument"]["TEM"]["acquisition_mode"] = (
@@ -534,18 +549,28 @@ def get_metadata(self):
             metadata["Acquisition_instrument"]["TEM"]["camera_length"] = (
                 original_metadata["Microscope Info"].get("STEM Camera Length", 0)
             )
-            metadata["Acquisition_instrument"] = original_metadata["Microscope Info"]
-        metadata["Acquisition_instrument"] = {}
         return metadata, original_metadata
 
     def update_metadata(
         self, metadata=None, signal_dimensions=None, navigation_dimensions=None
     ):
         """
         Update the metadata for the image.
+
+        Parameters
+        ----------
+        metadata : dict, optional
+            The metadata to update.
+        signal_dimensions : int, optional
+            The number of signal dimensions.
+        navigation_dimensions : int, optional
+            The number of navigation dimensions.
         """
         if metadata is None:
             metadata = {}
+        if navigation_dimensions is None and signal_dimensions is None:
+            signal_dimensions = self.ndim
+            navigation_dimensions = 0
         formatted_metadata = {}
 
         formatted_metadata["Acquisition"] = {}
@@ -569,26 +594,22 @@ def update_metadata(
         if navigation_dimensions > 0:
             formatted_metadata["Meta Data"]["IsSequence"] = "true"
         dict2group(formatted_metadata, self.image_tags)
+
+        # Update Microscope Info
+        if "Acquisition_instrument" in metadata and "TEM" in metadata["Acquisition_instrument"]:
+            self.image_tags.create_group("Microscope Info")
+            microscope_info_dict = {"Voltage": metadata["Acquisition_instrument"]["TEM"].get("beam_energy",0) * 1000,
+            "Illumination Mode": metadata["Acquisition_instrument"]["TEM"].get("acquisition_mode", "Unknown"),
+            "Indicated Magnification": metadata["Acquisition_instrument"]["TEM"].get("magnification", 0),
+            "STEM Camera Length": metadata["Acquisition_instrument"]["TEM"].get("camera_length", 0)}
+
+
+            dict2group(microscope_info_dict, self.image_tags["Microscope Info"])
+
         self.image_tags.create_group("UserTags")
         dict2group(metadata, self.image_tags["UserTags"])
         return
 
-    def to_signal_dict(self):
-        """
-        Convert the image to a Hyperspy signal dictionary.
-        """
-        data = self.get_data()
-        metadata, original_metadata = self.get_metadata()
-        axes = []
-        for axis in range(len(data.shape)):
-            axes.append(self.get_axis_dict(axis))
-        return {
-            "data": data,
-            "metadata": metadata,
-            "original_metadata": original_metadata,
-            "axes": axes,
-        }
-
 
 def dict2group(dictionary, group):
     for key, value in dictionary.items():

rsciio/tests/test_dm5.py

@@ -83,3 +83,38 @@ def test_save_load_files(
             assert "nm" in s.axes_manager[i].units
             assert s.axes_manager[i].scale == 0.1
             assert s.axes_manager[i].size == int(original[i][-2:])
+
+    def test_save_load_undefined_axes(
+        self, tmp_path
+    ):
+        fname = tmp_path/ f"test_save_undefined.dm5"
+
+        data_shape = [10, 11, 12, 13]
+        data = np.ones(data_shape, dtype=np.float32)
+        signal = hs.signals.Signal2D(data)
+        signal.save(fname, overwrite=True)
+        s = hs.load(fname)
+        for i in range(4):
+            assert s.axes_manager[i].name == ""
+            assert s.axes_manager[i].units == ""
+
+    def test_save_load_metadata(
+        self, tmp_path
+    ):
+        fname = tmp_path/ f"test_save_undefined.dm5"
+
+        data_shape = [10, 11, 12, 13]
+        data = np.ones(data_shape, dtype=np.float32)
+        signal = hs.signals.Signal2D(data)
+        signal.metadata.General.title = "test"
+        signal.metadata.add_node("Acquisition_instrument.TEM")
+        signal.metadata.Acquisition_instrument.TEM.beam_energy  = 200
+        signal.metadata.Acquisition_instrument.TEM.magnification = 100
+        signal.metadata.Acquisition_instrument.TEM.camera_length = 10
+
+        signal.save(fname, overwrite=True)
+        s = hs.load(fname)
+        assert  s.metadata.Acquisition_instrument.TEM.beam_energy == 200
+        assert  s.metadata.Acquisition_instrument.TEM.camera_length == 10
+        assert  s.metadata.Acquisition_instrument.TEM.magnification == 100
+