Merge "Cleanup BaseMonitor unit tests"

[yardstick.git] / tests / unit / __init__.py

diff --git a/tests/unit/__init__.py b/tests/unit/__init__.py
index 5935abb..95b2b8a 100644 (file)
--- a/tests/unit/__init__.py
+++ b/tests/unit/__init__.py
@@ -12,9 +12,15 @@
 # See the License for the specific language governing permissions and\r
 # limitations under the License.\r
 \r
-from yardstick import tests\r
+import sys\r
+\r
+import mock\r
 \r
+from yardstick import tests\r
 \r
 # NOTE(ralonsoh): to be removed. Replace all occurrences of\r
 # tests.unit.STL_MOCKS with yardstick.tests.STL_MOCKS\r
 STL_MOCKS = tests.STL_MOCKS\r
+\r
+mock_stl = mock.patch.dict(sys.modules, tests.STL_MOCKS)\r
+mock_stl.start()\r