@@ -185,6 +185,7 @@ def test_sensitivity(
185
185
assert "k2" in sens_std .columns
186
186
assert np .allclose (4.732277 , sens_std ["k2" ].sum (), atol = 0.00001 )
187
187
188
+
188
189
def test_validate_unique_column_names_passes () -> None :
189
190
"""Test that validation passes with unique column names."""
190
191
df = pd .DataFrame (columns = ["A" , "B" , "C" ])
@@ -194,21 +195,21 @@ def test_validate_unique_column_names_passes() -> None:
194
195
def test_validate_unique_column_names_fails_single_duplicate () -> None :
195
196
"""Test that validation fails with a single duplicated column."""
196
197
df = pd .DataFrame (columns = ["A" , "B" , "A" ])
197
-
198
+
198
199
with pytest .raises (ValueError ) as exc_info :
199
200
ms .ModelSolver ._validate_unique_column_names (df )
200
-
201
+
201
202
assert "Found duplicate column names in DataFrame" in str (exc_info .value )
202
203
assert "{'A': 2}" in str (exc_info .value )
203
204
204
205
205
206
def test_validate_unique_column_names_fails_multiple_duplicates () -> None :
206
207
"""Test that validation fails with multiple duplicated columns."""
207
208
df = pd .DataFrame (columns = ["A" , "B" , "A" , "B" , "C" ])
208
-
209
+
209
210
with pytest .raises (ValueError ) as exc_info :
210
211
ms .ModelSolver ._validate_unique_column_names (df )
211
-
212
+
212
213
assert "Found duplicate column names in DataFrame" in str (exc_info .value )
213
214
# Check both duplicates are reported
214
215
assert "'A': 2" in str (exc_info .value )
@@ -220,4 +221,4 @@ def test_validate_unique_column_names_empty_df() -> None:
220
221
df = pd .DataFrame ()
221
222
with pytest .raises (ValueError ) as exc_info :
222
223
ms .ModelSolver ._validate_unique_column_names (df )
223
- assert "DataFrame has no columns" in str (exc_info )
224
+ assert "DataFrame has no columns" in str (exc_info )
0 commit comments