update: split tests

tests/test_load_optimizers.py

@@ -38,24 +38,3 @@ def test_load_optimizers_valid(valid_optimizer_names):
 def test_load_optimizers_invalid(invalid_optimizer_names):
  with pytest.raises(NotImplementedError):
  load_optimizers(invalid_optimizer_names)
-
-
-@pytest.mark.parametrize('optimizer_names', VALID_OPTIMIZER_NAMES)
-def test_learning_rate(optimizer_names):
- with pytest.raises(ValueError):
- optimizer = load_optimizers(optimizer_names)
- optimizer(None, lr=-1e-2)
-
-
-@pytest.mark.parametrize('optimizer_names', VALID_OPTIMIZER_NAMES)
-def test_epsilon(optimizer_names):
- with pytest.raises(ValueError):
- optimizer = load_optimizers(optimizer_names)
- optimizer(None, eps=-1e-6)
-
-
-@pytest.mark.parametrize('optimizer_names', VALID_OPTIMIZER_NAMES)
-def test_weight_decay(optimizer_names):
- with pytest.raises(ValueError):
- optimizer = load_optimizers(optimizer_names)
- optimizer(None, weight_decay=-1e-3)

tests/test_optimizer_parameters.py

@@ -0,0 +1,41 @@
+from typing import List
+
+import pytest
+
+from pytorch_optimizer import load_optimizers
+
+VALID_OPTIMIZER_NAMES: List[str] = [
+ 'adamp',
+ 'sgdp',
+ 'madgrad',
+ 'ranger',
+ 'ranger21',
+ 'radam',
+ 'adabound',
+ 'adahessian',
+ 'adabelief',
+ 'diffgrad',
+ 'diffrgrad',
+ 'lamb',
+]
+
+
+@pytest.mark.parametrize('optimizer_names', VALID_OPTIMIZER_NAMES)
+def test_learning_rate(optimizer_names):
+ with pytest.raises(ValueError):
+ optimizer = load_optimizers(optimizer_names)
+ optimizer(None, lr=-1e-2)
+
+
+@pytest.mark.parametrize('optimizer_names', VALID_OPTIMIZER_NAMES)
+def test_epsilon(optimizer_names):
+ with pytest.raises(ValueError):
+ optimizer = load_optimizers(optimizer_names)
+ optimizer(None, eps=-1e-6)
+
+
+@pytest.mark.parametrize('optimizer_names', VALID_OPTIMIZER_NAMES)
+def test_weight_decay(optimizer_names):
+ with pytest.raises(ValueError):
+ optimizer = load_optimizers(optimizer_names)
+ optimizer(None, weight_decay=-1e-3)