Skip to content

Commit 71958d9

Browse files
committed
rnmd process_imbalance_sampler to verify_imbalance_sampler; tests passed
1 parent 111ff0a commit 71958d9

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

unittests/test_model.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)