diff --git a/python-package/tests/__init__.py b/python-package/tests/__init__.py index e69de29..d317a1f 100644 --- a/python-package/tests/__init__.py +++ b/python-package/tests/__init__.py @@ -0,0 +1,4 @@ +import logging + +test_logger = logging.getLogger("test") +test_logger.setLevel(logging.INFO) diff --git a/python-package/tests/azure/create_azure_resources.py b/python-package/tests/azure/create_azure_resources.py index 6c499de..10fa240 100644 --- a/python-package/tests/azure/create_azure_resources.py +++ b/python-package/tests/azure/create_azure_resources.py @@ -2,6 +2,7 @@ import pathlib import mc2client as mc2 +from . import test_logger as logger tests_dir = pathlib.Path(__file__).parent.absolute() config = os.path.join(tests_dir, "../test.yaml") @@ -9,17 +10,17 @@ dummy_file = os.path.join(tests_dir, "dummy.txt") -print("Creating resource group") +logger.info("Creating resource group") mc2.create_resource_group() -print("Creating storage") +logger.info("Creating storage") mc2.create_storage() -print("Creating container") +logger.info("Creating container") mc2.create_container() -print("Uploading file") +logger.info("Uploading file") mc2.upload_file(dummy_file, "dummy.txt") -print("Creating cluster") +logger.info("Creating cluster") mc2.create_cluster() diff --git a/python-package/tests/azure/delete_azure_resources.py b/python-package/tests/azure/delete_azure_resources.py index cd3c58c..39d4425 100644 --- a/python-package/tests/azure/delete_azure_resources.py +++ b/python-package/tests/azure/delete_azure_resources.py @@ -2,30 +2,32 @@ import pathlib import mc2client as mc2 +from . import test_logger as logger -print("Setting config file path") + +logger.info("Setting config file path") tests_dir = pathlib.Path(__file__).parent.absolute() config = os.path.join(tests_dir, "../test.yaml") mc2.set_config(config) download_path = os.path.join(tests_dir, "dummy_downloaded.txt") -print("Downloading file") +logger.info("Downloading file") mc2.download_file("dummy.txt", download_path) if not os.path.isfile(download_path): - print("Error: Couldn't download file from Azure") + logger.info("Error: Couldn't download file from Azure") else: os.remove(download_path) -print("Deleting container") +logger.info("Deleting container") mc2.delete_container() -print("Deleting storage") +logger.info("Deleting storage") mc2.delete_storage() -print("Deleting cluster") +logger.info("Deleting cluster") mc2.delete_cluster() -print("Deleting resource group") +logger.info("Deleting resource group") mc2.delete_resource_group()