From 5d92d1ab34553ef96371b154baf76545a8c01eba Mon Sep 17 00:00:00 2001 From: Gawtham Senthilvelan Date: Mon, 23 Jun 2025 14:34:20 -0400 Subject: [PATCH] Error Checking and Changing Expected Answer File Name --- featurize_assignment_test.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/featurize_assignment_test.py b/featurize_assignment_test.py index 887fa59..90826fe 100644 --- a/featurize_assignment_test.py +++ b/featurize_assignment_test.py @@ -9,11 +9,15 @@ @pytest.fixture(scope="session") def get_namespace(): - script_fname = "featurize_assignment_ans.py" + script_fname = "featurize_assignment.py" script_content = open(script_fname).read() namespace = {} exec(script_content, namespace) + required_vars = ["ax_client", "X_train", "y_train", "X_candidate", "optimal_composition", "max_hardness", "most_important", "correlation", "n_hard_materials"] + missing_vars = [var for var in required_vars if var not in namespace] + if missing_vars: + pytest.skip(f"Assignment incomplete. Missing variables: {missing_vars}") return namespace