@@ -827,13 +827,17 @@ def timer_callback(self, t=beam.DoFn.TimestampParam):
827
827
assert_that (actual , equal_to (expected ))
828
828
829
829
def test_pardo_et_timer_with_no_firing (self ):
830
- if type (self ) in [FnApiRunnerTest ,
831
- FnApiRunnerTestWithGrpc ,
832
- FnApiRunnerTestWithGrpcAndMultiWorkers ,
833
- FnApiRunnerTestWithDisabledCaching ,
834
- FnApiRunnerTestWithMultiWorkers ,
835
- FnApiRunnerTestWithBundleRepeat ,
836
- FnApiRunnerTestWithBundleRepeatAndMultiWorkers ]:
830
+ if type (self ).__name__ in {
831
+ 'FnApiRunnerTest' ,
832
+ 'FnApiRunnerTestWithGrpc' ,
833
+ 'FnApiRunnerTestWithGrpcAndMultiWorkers' ,
834
+ 'FnApiRunnerTestWithDisabledCaching' ,
835
+ 'FnApiRunnerTestWithMultiWorkers' ,
836
+ 'FnApiRunnerTestWithBundleRepeat' ,
837
+ 'FnApiRunnerTestWithBundleRepeatAndMultiWorkers' ,
838
+ 'SamzaRunnerTest' ,
839
+ 'SparkRunnerTest'
840
+ }:
837
841
raise unittest .SkipTest ("https://github.com/apache/beam/issues/35168" )
838
842
839
843
# The timer will not fire. It is initially set to T + 10, but then it is
@@ -842,20 +846,36 @@ def test_pardo_et_timer_with_no_firing(self):
842
846
self ._run_pardo_et_timer_test (5 , 10 , True , True , [])
843
847
844
848
def test_pardo_et_timer_with_no_reset (self ):
845
- if type (self ) in [FnApiRunnerTest ,
846
- FnApiRunnerTestWithGrpc ,
847
- FnApiRunnerTestWithGrpcAndMultiWorkers ,
848
- FnApiRunnerTestWithDisabledCaching ,
849
- FnApiRunnerTestWithMultiWorkers ,
850
- FnApiRunnerTestWithBundleRepeat ,
851
- FnApiRunnerTestWithBundleRepeatAndMultiWorkers ]:
849
+ if type (self ).__name__ in {
850
+ 'FnApiRunnerTest' ,
851
+ 'FnApiRunnerTestWithGrpc' ,
852
+ 'FnApiRunnerTestWithGrpcAndMultiWorkers' ,
853
+ 'FnApiRunnerTestWithDisabledCaching' ,
854
+ 'FnApiRunnerTestWithMultiWorkers' ,
855
+ 'FnApiRunnerTestWithBundleRepeat' ,
856
+ 'FnApiRunnerTestWithBundleRepeatAndMultiWorkers' ,
857
+ 'SamzaRunnerTest' ,
858
+ 'SparkRunnerTest'
859
+ }:
852
860
raise unittest .SkipTest ("https://github.com/apache/beam/issues/35168" )
853
861
854
862
# The timer will not fire. It is initially set to T + 10, and then it is
855
863
# cleared at T + 4 and never set again (count is not reset).
856
864
self ._run_pardo_et_timer_test (5 , 10 , False , True , [])
857
865
858
866
def test_pardo_et_timer_with_no_reset_and_no_clear (self ):
867
+ if type (self ).__name__ in {
868
+ 'FnApiRunnerTest' ,
869
+ 'FnApiRunnerTestWithGrpc' ,
870
+ 'FnApiRunnerTestWithGrpcAndMultiWorkers' ,
871
+ 'FnApiRunnerTestWithDisabledCaching' ,
872
+ 'FnApiRunnerTestWithMultiWorkers' ,
873
+ 'FnApiRunnerTestWithBundleRepeat' ,
874
+ 'FnApiRunnerTestWithBundleRepeatAndMultiWorkers' ,
875
+ 'SamzaRunnerTest' ,
876
+ 'SparkRunnerTest'
877
+ }:
878
+ raise unittest .SkipTest ("https://github.com/apache/beam/issues/35168" )
859
879
# The timer will fire at T + 10. After the timer is set, it is never
860
880
# cleared or set again.
861
881
self ._run_pardo_et_timer_test (5 , 10 , False , False , ["fired" ])
0 commit comments