Skip to content

Commit 2b2b04b

Browse files
committed
Update tests
1 parent 498e7a5 commit 2b2b04b

File tree

5 files changed

+7
-7
lines changed

5 files changed

+7
-7
lines changed

tests/test_MhaMlpClassifier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def classifier():
2727
act_names="ReLU",
2828
dropout_rates=0.2,
2929
optim="BaseGA",
30-
optim_paras={"epoch": 10, "pop_size": 20},
30+
optim_params={"epoch": 10, "pop_size": 20},
3131
seed=42,
3232
verbose=False
3333
)

tests/test_MhaMlpRegressor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def regressor():
2727
act_names="ReLU",
2828
dropout_rates=0.2,
2929
optim="BaseGA",
30-
optim_paras={"epoch": 10, "pop_size": 20},
30+
optim_params={"epoch": 10, "pop_size": 20},
3131
seed=42,
3232
verbose=False
3333
)

tests/test_MhaMlpTuner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def param_dict():
2828
'act_names': ['Tanh', 'ELU'],
2929
'dropout_rates': [None],
3030
'optim': ['BaseGA'],
31-
'optim_paras': [
31+
'optim_params': [
3232
{"epoch": 10, "pop_size": 20},
3333
{"epoch": 20, "pop_size": 20},
3434
],

tests/test_MlpClassifier.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,12 @@ def test_initialization(create_mlp_classifier):
3636
assert clf.seed == 42
3737

3838

39-
def test_process_data(create_mlp_classifier, generate_data):
39+
def test__process_data(create_mlp_classifier, generate_data):
4040
"""Test data processing method with validation split."""
4141
clf = create_mlp_classifier
4242
X, y = generate_data
4343

44-
train_loader, X_valid_tensor, y_valid_tensor = clf.process_data(X, y)
44+
train_loader, X_valid_tensor, y_valid_tensor = clf._process_data(X, y)
4545
assert train_loader is not None
4646
assert X_valid_tensor is not None
4747
assert y_valid_tensor is not None

tests/test_MlpRegressor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ def test_initialization(model):
2929
assert model.verbose is False
3030

3131

32-
def test_process_data(model):
32+
def test__process_data(model):
3333
"""Test the data processing and tensor conversion."""
34-
train_loader, X_valid_tensor, y_valid_tensor = model.process_data(X, y)
34+
train_loader, X_valid_tensor, y_valid_tensor = model._process_data(X, y)
3535

3636
# Check the training loader data format
3737
for batch_X, batch_y in train_loader:

0 commit comments

Comments
 (0)