Support almost equals for assertMeasured check.

PiperOrigin-RevId: 766756498

Author: OpenHTF Owners

openhtf/util/test.py

@@ -925,16 +925,38 @@ def _check_phase(phase_record, strict=False):
         _check_phase(phase_record)
 
   @_assert_phase_or_test_record
-  def assertMeasured(self, phase_record, measurement, value=mock.ANY):
+  def assertMeasured(
+      self, phase_record, measurement, value=mock.ANY, delta=None
+  ):
     self.assertTrue(
         phase_record.measurements[measurement].measured_value.is_value_set,
-        'Measurement %s not set' % measurement)
+        'Measurement %s not set' % measurement,
+    )
     if value is not mock.ANY:
+      if isinstance(value, float):
+        self.assertAlmostEqual(
+            value,
+            phase_record.measurements[measurement].measured_value.value,
+            delta=delta,
+            'Measurement %s has wrong value: expected %s, got %s, tolerance %s'
+            % (
+                measurement,
+                value,
+                phase_record.measurements[measurement].measured_value.value,
+            ),
+            delta,
+        )
+    else:
       self.assertEqual(
-          value, phase_record.measurements[measurement].measured_value.value,
-          'Measurement %s has wrong value: expected %s, got %s' %
-          (measurement, value,
-           phase_record.measurements[measurement].measured_value.value))
+          value,
+          phase_record.measurements[measurement].measured_value.value,
+          'Measurement %s has wrong value: expected %s, got %s'
+          % (
+              measurement,
+              value,
+              phase_record.measurements[measurement].measured_value.value,
+          ),
+      )
 
   @_assert_phase_or_test_record
   def assertMeasurementPass(self, phase_record, measurement, value=mock.ANY):