Skip to content

Commit a2a1690

Browse files
clean up unused part of cloudpickle_pickler_test
1 parent 9192f18 commit a2a1690

File tree

1 file changed

+5
-16
lines changed

1 file changed

+5
-16
lines changed

sdks/python/apache_beam/internal/cloudpickle_pickler_test.py

Lines changed: 5 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -236,24 +236,13 @@ def sample_func():
236236

237237
code_obj = sample_func.__code__
238238
original_filename = os.path.abspath(code_obj.co_filename)
239+
pickled_code = beam_cloudpickle.dumps(code_obj)
240+
unpickled_code = beam_cloudpickle.loads(pickled_code)
239241

240-
try:
241-
pickled_code = beam_cloudpickle.dumps(code_obj)
242-
unpickled_code = beam_cloudpickle.loads(pickled_code)
242+
mock_filepath_interceptor.assert_called()
243243

244-
mock_filepath_interceptor.assert_called()
245-
246-
unpickled_filename = os.path.abspath(unpickled_code.co_filename)
247-
self.assertEqual(unpickled_filename, original_filename)
248-
249-
except AttributeError as e:
250-
if 'get_code_object_params' in str(e):
251-
self.fail(
252-
"Vendored cloudpickle BUG: AttributeError "
253-
f"'get_code_object_params' raised during CodeType pickling. "
254-
f"Error: {e}")
255-
else:
256-
raise
244+
unpickled_filename = os.path.abspath(unpickled_code.co_filename)
245+
self.assertEqual(unpickled_filename, original_filename)
257246

258247

259248
if __name__ == '__main__':

0 commit comments

Comments
 (0)