Update test_selection.py

test_selection.py

@@ -1,4 +1,3 @@
-# Define costs for tests in each category
 TEST_PRICES = {
     "Haematology": {
         "Haemoglobin estimation": 50,
@@ -43,7 +42,7 @@ TEST_PRICES = {
 
 def get_tests_by_category(categories):
     if not categories:
-        return []  # Return an empty list if no categories are selected
+        return []
     
     available_tests = []
     for category in categories:
@@ -59,7 +58,7 @@ def select_tests(patient_id, selected_tests, data):
     total_cost = 0
     tests_selected = []
     for item in selected_tests:
-        test_name = item.split(" - ")[0]  # Extract the test name
+        test_name = item.split(" - ")[0]
         for category, tests in TEST_PRICES.items():
             if test_name in tests:
                 total_cost += tests[test_name]