@@ -244,9 +244,9 @@ def test_child_module_logger_can_override_logging_level(self, mock_emit):
244244 reason = '[interactive] dependency is not installed.' )
245245class ProgressIndicatorTest (unittest .TestCase ):
246246 def setUp (self ):
247- self .patcher = patch (
248- 'apache_beam.runners.interactive.cache_manager.CacheManager.cleanup ' )
249- self .patcher .start ()
247+ self .gcs_patcher = patch (
248+ 'apache_beam.io.gcp.gcsfilesystem.GCSFileSystem.delete ' )
249+ self .gcs_patcher .start ()
250250 ie .new_env ()
251251
252252 @patch ('IPython.get_ipython' , new_callable = mock_get_ipython )
@@ -283,7 +283,7 @@ def test_progress_in_HTML_JS_when_in_notebook(
283283 mocked_js .assert_called ()
284284
285285 def tearDown (self ):
286- self .patcher .stop ()
286+ self .gcs_patcher .stop ()
287287
288288
289289@unittest .skipIf (
@@ -293,9 +293,9 @@ class MessagingUtilTest(unittest.TestCase):
293293 SAMPLE_DATA = {'a' : [1 , 2 , 3 ], 'b' : 4 , 'c' : '5' , 'd' : {'e' : 'f' }}
294294
295295 def setUp (self ):
296- self .patcher = patch (
297- 'apache_beam.runners.interactive.cache_manager.CacheManager.cleanup ' )
298- self .patcher .start ()
296+ self .gcs_patcher = patch (
297+ 'apache_beam.io.gcp.gcsfilesystem.GCSFileSystem.delete ' )
298+ self .gcs_patcher .start ()
299299 ie .new_env ()
300300
301301 def test_as_json_decorator (self ):
@@ -308,7 +308,7 @@ def dummy():
308308 self .assertEqual (json .loads (dummy ()), MessagingUtilTest .SAMPLE_DATA )
309309
310310 def tearDown (self ):
311- self .patcher .stop ()
311+ self .gcs_patcher .stop ()
312312
313313
314314class GeneralUtilTest (unittest .TestCase ):
0 commit comments