From ce0f9c4ebf830eee9247c85aa94b34eecc9abc26 Mon Sep 17 00:00:00 2001 From: "Brian C. Lane" Date: Wed, 5 Feb 2020 11:02:29 -0800 Subject: [PATCH] tests: Use unittest.mock Related: rhbz#1785154 --- tests/pylorax/test_recipes.py | 2 +- tests/pylorax/test_workspace.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pylorax/test_recipes.py b/tests/pylorax/test_recipes.py index f08373ee..62d5b57e 100644 --- a/tests/pylorax/test_recipes.py +++ b/tests/pylorax/test_recipes.py @@ -15,11 +15,11 @@ # along with this program. If not, see . # import os -import mock from pytoml import TomlError import shutil import tempfile import unittest +from unittest import mock import pylorax.api.recipes as recipes from pylorax.api.compose import add_customizations, customize_ks_template diff --git a/tests/pylorax/test_workspace.py b/tests/pylorax/test_workspace.py index c6b950ba..7f8517fe 100644 --- a/tests/pylorax/test_workspace.py +++ b/tests/pylorax/test_workspace.py @@ -15,10 +15,10 @@ # along with this program. If not, see . # import os -import mock import shutil import tempfile import unittest +from unittest import mock import pylorax.api.recipes as recipes from pylorax.api.workspace import workspace_dir, workspace_read, workspace_write, workspace_delete