@@ -925,13 +925,13 @@ def single_class_data():
925925 return X_train , y_train
926926
927927
928- def test_process_imbalance_sampler_imbalanced_data (
928+ def test_verify_imbalance_sampler_imbalanced_data (
929929 capfd , imbalanced_data , model_with_sampler
930930):
931- """Test process_imbalance_sampler with imbalanced data."""
931+ """Test verify_imbalance_sampler with imbalanced data."""
932932 X_train , y_train = imbalanced_data
933933
934- model_with_sampler .process_imbalance_sampler (X_train , y_train )
934+ model_with_sampler .verify_imbalance_sampler (X_train , y_train )
935935
936936 # Capture printed output
937937 captured = capfd .readouterr ()
@@ -941,13 +941,11 @@ def test_process_imbalance_sampler_imbalanced_data(
941941 assert "1 12" in captured .out , "Should print correct count for minority class."
942942
943943
944- def test_process_imbalance_sampler_balanced_data (
945- capfd , balanced_data , model_with_sampler
946- ):
947- """Test process_imbalance_sampler with already balanced data."""
944+ def test__balanced_data (capfd , balanced_data , model_with_sampler ):
945+ """Test verify_imbalance_sampler with already balanced data."""
948946 X_train , y_train = balanced_data
949947
950- model_with_sampler .process_imbalance_sampler (X_train , y_train )
948+ model_with_sampler .verify_imbalance_sampler (X_train , y_train )
951949
952950 # Capture printed output
953951 captured = capfd .readouterr ()
@@ -958,14 +956,14 @@ def test_process_imbalance_sampler_balanced_data(
958956 assert "1 2" in captured .out , "Should print correct count for minority class."
959957
960958
961- def test_process_imbalance_sampler_single_class_data (
959+ def test_verify_imbalance_sampler_single_class_data (
962960 single_class_data , model_with_sampler
963961):
964- """Test process_imbalance_sampler with single-class data."""
962+ """Test verify_imbalance_sampler with single-class data."""
965963 X_train , y_train = single_class_data
966964
967965 with pytest .raises (ValueError ):
968- model_with_sampler .process_imbalance_sampler (X_train , y_train )
966+ model_with_sampler .verify_imbalance_sampler (X_train , y_train )
969967
970968
971969@pytest .fixture
0 commit comments