fix: correct index column reference in validation function

Author: bencap

src/mavedb/lib/validation/dataframe/calibration.py

@@ -80,7 +80,7 @@ def validate_and_standardize_calibration_classes_dataframe(
             validate_data_column(standardized_classes_df[c], force_numeric=False)
             validate_calibration_classes(calibration, standardized_classes_df[c])
 
-        if c == index_column:
+        if column_mapping[c] == index_column:
             validate_index_existence_in_score_set(
                 db, score_set, standardized_classes_df[column_mapping[c]], column_mapping[c]
             )
@@ -150,9 +150,6 @@ def validate_index_existence_in_score_set(
     Returns:
         None: Function returns nothing if validation passes.
     """
-    print(index_column.tolist())
-    print(index_column_name)
-
     if index_column_name.lower() == calibration_variant_column_name:
         existing_resources = set(
             db.scalars(

tests/validation/dataframe/test_calibration.py

@@ -185,7 +185,9 @@ def test_validate_and_standardize_calibration_classes_dataframe_drops_null_class
 
         assert result.equals(expected_df)
 
-    def test_validate_and_standardize_calibration_classes_dataframe_invalid_variants(self, mock_dependencies):
+    def test_validate_and_standardize_calibration_classes_dataframe_propagates_nonexistent_variants(
+        self, mock_dependencies
+    ):
         """Test ValidationError when variant URN validation fails."""
         mock_db = Mock()
         mock_score_set = Mock()
@@ -204,6 +206,14 @@ def test_validate_and_standardize_calibration_classes_dataframe_invalid_variants
         mock_scalars.all.return_value = []
         mock_db.scalars.return_value = mock_scalars
 
+        mock_classification1 = Mock()
+        mock_classification1.class_ = "A"
+        mock_calibration.functional_classifications = [mock_classification1]
+
+        mock_dependencies["validate_index_existence_in_score_set"].side_effect = ValidationError(
+            "The following resources do not exist in the score set: var1"
+        )
+
         with pytest.raises(ValidationError, match="The following resources do not exist in the score set: var1"):
             validate_and_standardize_calibration_classes_dataframe(mock_db, mock_score_set, mock_calibration, input_df)