refactor: test_ranger_parameters

Author: kozistr

tests/test_general_optimizer_parameters.py

@@ -84,13 +84,6 @@ def test_momentum(optimizer_name):
         optimizer(None, momentum=-1e-3)
 
 
-@pytest.mark.parametrize('optimizer_name', ['ranger'])
-def test_lookahead_k(optimizer_name):
-    optimizer = load_optimizer(optimizer_name)
-    with pytest.raises(ValueError):
-        optimizer(None, k=-1)
-
-
 @pytest.mark.parametrize('optimizer_name', ['nero', 'apollo', 'sm3', 'msvag', 'ranger21'])
 def test_beta(optimizer_name):
     optimizer = load_optimizer(optimizer_name)

tests/test_optimizer_parameters.py

@@ -222,3 +222,15 @@ def test_apollo_parameters():
     # test weight_decay_type
     with pytest.raises(ValueError):
         opt(None, weight_decay_type='dummy')
+
+
+def test_ranger_parameters():
+    opt = load_optimizer('ranger')
+
+    # test ema ratio `alpha`
+    with pytest.raises(ValueError):
+        opt(None, alpha=-0.1)
+
+    # test lookahead step `k`
+    with pytest.raises(ValueError):
+        opt(None, k=-1)