[BEAM-13955] Fix pylint breakage from #16836 (#16867)

apache · Feb 16, 2022 · 9cac23a · 9cac23a
1 parent 6e98dd4
commit 9cac23a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sdks/python/apache_beam/transforms/userstate_test.py b/sdks/python/apache_beam/transforms/userstate_test.py
@@ -164,29 +164,29 @@ 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)
 
     context = pipeline_context.PipelineContext()
     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)