Skip to content

Commit 6393091

Browse files
committed
update: test cases
1 parent 8902016 commit 6393091

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/test_load_modules.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,17 @@ def test_load_lr_scheduler_invalid(invalid_lr_scheduler_names):
3939

4040
def test_get_supported_optimizers():
4141
assert len(get_supported_optimizers()) == 76
42+
assert len(get_supported_optimizers('adam*')) == 7
43+
assert len(get_supported_optimizers(['adam*', 'ranger*'])) == 9
4244

4345

4446
def test_get_supported_lr_schedulers():
4547
assert len(get_supported_lr_schedulers()) == 16
48+
assert len(get_supported_lr_schedulers('cosine*')) == 4
49+
assert len(get_supported_lr_schedulers(['cosine*', '*warm*'])) == 5
4650

4751

4852
def test_get_supported_loss_functions():
4953
assert len(get_supported_loss_functions()) == 13
54+
assert len(get_supported_loss_functions('*focal*')) == 4
55+
assert len(get_supported_loss_functions(['*focal*', 'bce*'])) == 5

0 commit comments

Comments
 (0)