From 85342980c14d51d58e27dbb3eb494f55f4a8494b Mon Sep 17 00:00:00 2001 From: Max Wang Date: Fri, 8 Mar 2024 13:30:00 -0500 Subject: [PATCH] Update test names to include expected output --- .../scripts/generate_suite_for_demo.py | 26 ++++++++++--------- 1 file changed, 14 insertions(+), 12 deletions(-) diff --git a/src/translator_testing_model/scripts/generate_suite_for_demo.py b/src/translator_testing_model/scripts/generate_suite_for_demo.py index 5b6dd4b..b022640 100644 --- a/src/translator_testing_model/scripts/generate_suite_for_demo.py +++ b/src/translator_testing_model/scripts/generate_suite_for_demo.py @@ -26,14 +26,26 @@ def create_test_assets_from_tsv(test_assets): for row in test_assets: if row.get("Relationship") == "": continue + expected_output = None + if row.get("Expected Result / Suggested Comparator") == "4_NeverShow": + expected_output = "NeverShow" + elif row.get("Expected Result / Suggested Comparator") == "3_BadButForgivable": + expected_output = "BadButForgivable" + elif row.get("Expected Result / Suggested Comparator") == "2_Acceptable": + expected_output = "Acceptable" + elif row.get("Expected Result / Suggested Comparator") == "1_TopAnswer": + expected_output = "TopAnswer" + else: + print(f"{row.get('id')} has invalid expected output") + continue ta = TestAsset(id=row.get("id").replace(":", "_"), - name=row.get("OutputName").replace(" ", "_") + "_" + row.get("Relationship").lower() + "_" + row.get("InputName (user choice)").replace(" ", "_"), + name=expected_output + ': ' + row.get("OutputName").replace(" ", "_") + "_" + row.get("Relationship").lower() + "_" + row.get("InputName (user choice)").replace(" ", "_"), description=row.get("OutputName").replace(" ", "_") + "_" + row.get("Relationship").lower() + "_" + row.get("InputName (user choice)").replace(" ", "_"), input_id=row.get("InputID, node normalized"), predicate_name=row.get("Relationship").lower(), predicate_id="biolink:"+row.get("Relationship").lower(), output_id=row.get("OutputID"), - expected_output="NeverShow", + expected_output=expected_output, test_metadata=TestMetadata(id=1), ) ta.input_name = row.get("InputName (user choice)") @@ -50,16 +62,6 @@ def create_test_assets_from_tsv(test_assets): ta.test_metadata = tmd ta.output_name = row.get("OutputName") ta.runner_settings = [row.get("Settings").lower()] - if row.get("Expected Result / Suggested Comparator") == "4_NeverShow": - ta.expected_output = "NeverShow" - elif row.get("Expected Result / Suggested Comparator") == "3_BadButForgivable": - ta.expected_output = "BadButForgivable" - elif row.get("Expected Result / Suggested Comparator") == "2_Acceptable": - ta.expected_output = "Acceptable" - elif row.get("Expected Result / Suggested Comparator") == "1_TopAnswer": - ta.expected_output = "TopAnswer" - else: - ta.expected_output = "NeverShow" if row.get("Well Known") == "yes": ta.well_known = True