diff --git a/tests/connectors/test_train.py b/tests/connectors/test_train.py index 6cf8bd01..b4f77cd1 100644 --- a/tests/connectors/test_train.py +++ b/tests/connectors/test_train.py @@ -142,6 +142,7 @@ def test_classify_read_four_cols_error(mocker): ) def test_classify_write_logs_new_model_id_integration(caplog): + caplog.set_level(logging.INFO) df = pd.DataFrame({ 'Example': ['apple', 'banana'], 'Category': ['fruit', 'fruit'], @@ -1425,6 +1426,7 @@ def test_lookup_write_logs_new_model_id(caplog): """ Integration test for lookup model creation logging """ + caplog.set_level(logging.INFO) df = pd.DataFrame({ 'Key': ['apple', 'banana'], 'Value': ['fruit', 'fruit'] @@ -1535,6 +1537,7 @@ def test_standardize_write_logs_new_model_id(caplog): """ Integration test for standardize model creation logging """ + caplog.set_level(logging.INFO) df = pd.DataFrame({ 'Find': ['ASAP', 'ETA'], 'Replace': ['As Soon As Possible', 'Estimated Time of Arrival'], diff --git a/tests/recipes/test_recipes.py b/tests/recipes/test_recipes.py index b2e09ef5..172d9fe5 100644 --- a/tests/recipes/test_recipes.py +++ b/tests/recipes/test_recipes.py @@ -131,7 +131,7 @@ def test_recipe_by_production_version(): """ Test running a recipe using a model ID and production version """ - df = wrangles.recipe.run("a6bac9e7-2388-4347") + df = wrangles.recipe.run("e954717c-fb9c-4c47") assert ( len(df) == 20 and list(df.columns) == ["header"] @@ -154,10 +154,10 @@ def test_recipe_by_production_semantic_version(mocker): return_value={"recipe": "{}"} ) - wrangles.recipe.run("a6bac9e7-2388-4347:production") + wrangles.recipe.run("e954717c-fb9c-4c47:production") model_content.assert_called_once_with( - "a6bac9e7-2388-4347", + "e954717c-fb9c-4c47", "production-version-id" ) @@ -178,9 +178,9 @@ def test_recipe_by_production_semantic_version_falls_back_to_latest( return_value={"recipe": "{}"} ) - wrangles.recipe.run("a6bac9e7-2388-4347:production") + wrangles.recipe.run("e954717c-fb9c-4c47:production") - model_content.assert_called_once_with("a6bac9e7-2388-4347", None) + model_content.assert_called_once_with("e954717c-fb9c-4c47", None) assert "No production version exists, defaulting to latest version" in caplog.text @@ -188,7 +188,7 @@ def test_recipe_by_version_latest(): """ Test running a recipe using a model ID and latest version """ - df = wrangles.recipe.run("a6bac9e7-2388-4347:latest") + df = wrangles.recipe.run("e954717c-fb9c-4c47:latest") assert ( len(df) == 10 and list(df.columns) == ["header"] @@ -198,7 +198,7 @@ def test_recipe_by_latest_version(): """ Test running a recipe using a model ID and latest version """ - df = wrangles.recipe.run("02fc0c63-1294-415b") + df = wrangles.recipe.run("1b41d016-7129-4b66") assert ( len(df) == 15 and list(df.columns) == ["header"]