Support specifying a tolerance when using assertMeasured with floats.

PiperOrigin-RevId: 766756498

Author: OpenHTF Owners

openhtf/util/test.py

@@ -925,16 +925,45 @@ 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:
-      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))
+      if isinstance(value, float):
+        self.assertAlmostEqual(
+            value,
+            phase_record.measurements[measurement].measured_value.value,
+            delta=delta,
+            msg=(
+                'Measurement %s has wrong value: expected %s, got %s,'
+                ' tolerance %s'
+            )
+            % (
+                measurement,
+                value,
+                phase_record.measurements[measurement].measured_value.value,
+                delta,
+            ),
+        )
+      else:
+        if delta is not None:
+          raise ValueError(
+              'Delta is not supported when expected value is not a float.'
+          )
+        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,
+            ),
+        )
 
   @_assert_phase_or_test_record
   def assertMeasurementPass(self, phase_record, measurement, value=mock.ANY):