From 9cac23aecc24541be146197e8ad95da738ba2951 Mon Sep 17 00:00:00 2001 From: Brian Hulette Date: Wed, 16 Feb 2022 08:24:31 -0800 Subject: [PATCH] [BEAM-13955] Fix pylint breakage from #16836 (#16867) --- sdks/python/apache_beam/transforms/userstate_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/sdks/python/apache_beam/transforms/userstate_test.py b/sdks/python/apache_beam/transforms/userstate_test.py index 24e996e6393a..5bee3b56defd 100644 --- a/sdks/python/apache_beam/transforms/userstate_test.py +++ b/sdks/python/apache_beam/transforms/userstate_test.py @@ -164,7 +164,7 @@ def test_state_spec_proto_conversion(self): context = pipeline_context.PipelineContext() state = BagStateSpec('statename', VarIntCoder()) state_proto = state.to_runner_api(context) - self.assertEquals( + self.assertEqual( beam_runner_api_pb2.FunctionSpec(urn=common_urns.user_state.BAG.urn), state_proto.protocol) @@ -172,21 +172,21 @@ def test_state_spec_proto_conversion(self): state = CombiningValueStateSpec( 'statename', VarIntCoder(), TopCombineFn(10)) state_proto = state.to_runner_api(context) - self.assertEquals( + self.assertEqual( beam_runner_api_pb2.FunctionSpec(urn=common_urns.user_state.BAG.urn), state_proto.protocol) context = pipeline_context.PipelineContext() state = SetStateSpec('setstatename', VarIntCoder()) state_proto = state.to_runner_api(context) - self.assertEquals( + self.assertEqual( beam_runner_api_pb2.FunctionSpec(urn=common_urns.user_state.BAG.urn), state_proto.protocol) context = pipeline_context.PipelineContext() state = ReadModifyWriteStateSpec('valuestatename', VarIntCoder()) state_proto = state.to_runner_api(context) - self.assertEquals( + self.assertEqual( beam_runner_api_pb2.FunctionSpec(urn=common_urns.user_state.BAG.urn), state_proto.protocol)