Remove extra test and increase retries on the process timer tests to avoid flakiness

Author: ktalluri456

sdks/python/apache_beam/runners/worker/statesampler_test.py

@@ -210,19 +210,18 @@ def test_timer_sampler_operation(self):
         state_duration_ms * (1.0 + margin_of_error),
         f"Timer metric was too high: {value} ms.")
 
-  @retry(reraise=True, stop=stop_after_attempt(3))
+  @retry(reraise=True, stop=stop_after_attempt(30))
   # Patch the problematic function to return the correct timer spec
   @patch('apache_beam.transforms.userstate.get_dofn_specs')
   def test_do_operation_process_timer(self, mock_get_dofn_specs):
     fn = TimerDoFn()
-    # get_dofn_specs returns a tuple of (state_specs, timer_specs)
     mock_get_dofn_specs.return_value = ([], [fn.TIMER_SPEC])
 
     if not statesampler.FAST_SAMPLER:
       self.skipTest('DoOperation test requires FAST_SAMPLER')
 
     state_duration_ms = 200
-    margin_of_error = 0.50
+    margin_of_error = 0.75
 
     counter_factory = CounterFactory()
     sampler = statesampler.StateSampler(
@@ -278,17 +277,14 @@ def test_do_operation_process_timer(self, mock_get_dofn_specs):
         found_counter, f"Expected counter '{expected_name}' to be created.")
 
     actual_value = found_counter.value()
+    logging.info("Actual value %d", actual_value)
     self.assertGreater(
         actual_value, state_duration_ms * (1.0 - margin_of_error))
     self.assertLess(actual_value, state_duration_ms * (1.0 + margin_of_error))
 
-  @retry(reraise=True, stop=stop_after_attempt(3))
+  @retry(reraise=True, stop=stop_after_attempt(30))
   @patch('apache_beam.runners.worker.operations.userstate.get_dofn_specs')
   def test_do_operation_process_timer_with_exception(self, mock_get_dofn_specs):
-    """
-    Tests that an exception from a timer is propagated and that the
-    sampler still records the time spent until the exception.
-    """
     fn = ExceptionTimerDoFn()
     mock_get_dofn_specs.return_value = ([], [fn.TIMER_SPEC])