update mt with impedance units

Author: kujaku11

mtpy/core/mt_dataframe.py

@@ -755,7 +755,7 @@ def to_z_object(self, units="mt"):
                 self.dataframe.station == self.station, f"z_{comp}_model_error"
             ]
 
-        z_object = Z(z, z_err, self.frequency, z_model_err, unit=units)
+        z_object = Z(z, z_err, self.frequency, z_model_err, units=units)
 
         if (z == 0).all():
             for comp in ["xx", "xy", "yx", "yy"]:

mtpy/core/transfer_function/base.py

@@ -102,11 +102,16 @@ def __eq__(self, other):
 
         # loop over variables to make sure they are all the same.
         for var in list(self._dataset.data_vars):
-            try:
-                if not (self._dataset[var] == other._dataset[var]).all().data:
+            has_tf_str = f"_has_{var.replace('transfer_function', 'tf')}"
+            if getattr(self, has_tf_str):
+                if getattr(other, has_tf_str):
+                    if not np.allclose(
+                        self._dataset[var].data, other._dataset[var].data
+                    ):
+                        self.logger.info(f"Transfer functions {var} not equal")
+                        return False
+                else:
                     return False
-            except TypeError:
-                continue
         return True
 
     def __deepcopy__(self, memo):

tests/core/test_mt.py

@@ -558,11 +558,11 @@ def test_impedance_in_ohms(self):
         z_obj = self.m1.Z
         z_obj.units = "ohm"
 
-        self.assertEqual(z_obj, self.mt_df.to_z_object())
+        self.assertEqual(z_obj, self.mt_df.to_z_object(units="ohm"))
 
     def test_impedance_not_equal(self):
 
-        self.assertNotEqual(self.m1.Z, self.mt_df.to_z_object())
+        self.assertNotEqual(self.m1.Z, self.mt_df.to_z_object(units="mt"))
 
 
 # =============================================================================