Use magicmock

mantidproject · Dec 5, 2024 · fec408f · fec408f
1 parent 04c8802
commit fec408f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/workspacemanager_presenter_test.py b/tests/workspacemanager_presenter_test.py
@@ -2,7 +2,7 @@
 import unittest
 
 import mock
-from mock import call, patch
+from mock import call, patch, MagicMock
 import numpy as np
 
 from mantid.api import AnalysisDataService
@@ -438,7 +438,7 @@ def test_compose_error(self):
 
     def test_ensure_that_the_ads_observer_calls_clear_handle(self):
         presenter = WorkspaceManagerPresenter(self.view)
-        presenter.clear_handle = mock.Mock()
+        presenter.clear_handle = MagicMock()
         self.assertTrue(isinstance(presenter._ads_observer, MSliceADSObserver))
         presenter._ads_observer = MSliceADSObserver(
             presenter.delete_handle, presenter.clear_handle, presenter.rename_handle
@@ -451,7 +451,7 @@ def test_ensure_that_the_ads_observer_calls_clear_handle(self):
 
     def test_ensure_that_the_ads_observer_calls_delete_handle(self):
         presenter = WorkspaceManagerPresenter(self.view)
-        presenter.delete_handle = mock.Mock()
+        presenter.delete_handle = MagicMock()
         self.assertTrue(isinstance(presenter._ads_observer, MSliceADSObserver))
         presenter._ads_observer = MSliceADSObserver(
             presenter.delete_handle, presenter.clear_handle, presenter.rename_handle
@@ -464,7 +464,7 @@ def test_ensure_that_the_ads_observer_calls_delete_handle(self):
 
     def test_ensure_that_the_ads_observer_calls_rename_handle(self):
         presenter = WorkspaceManagerPresenter(self.view)
-        presenter.rename_handle = mock.Mock()
+        presenter.rename_handle = MagicMock()
         self.assertTrue(isinstance(presenter._ads_observer, MSliceADSObserver))
         presenter._ads_observer = MSliceADSObserver(
             presenter.delete_handle, presenter.clear_handle, presenter.rename_handle