refactor: test_apollo_parameters

Author: kozistr

tests/test_general_optimizer_parameters.py

@@ -70,22 +70,6 @@ def test_weight_decay(optimizer_name):
         optimizer(None, **config)
 
 
-@pytest.mark.parametrize('optimizer_name', ['apollo'])
-def test_weight_decay_type(optimizer_name):
-    optimizer = load_optimizer(optimizer_name)
-
-    with pytest.raises(ValueError):
-        optimizer(None, weight_decay_type='dummy')
-
-
-@pytest.mark.parametrize('optimizer_name', ['apollo'])
-def test_rebound(optimizer_name):
-    optimizer = load_optimizer(optimizer_name)
-
-    with pytest.raises(ValueError):
-        optimizer(None, rebound='dummy')
-
-
 @pytest.mark.parametrize('optimizer_name', ['adamp', 'sgdp'])
 def test_wd_ratio(optimizer_name):
     optimizer = load_optimizer(optimizer_name)

tests/test_optimizer_parameters.py

@@ -210,3 +210,15 @@ def test_lars_parameters():
     # test trust_coefficient
     with pytest.raises(ValueError):
         opt(None, trust_coefficient=-1e-3)
+
+
+def test_apollo_parameters():
+    opt = load_optimizer('apollo')
+
+    # test rebound type
+    with pytest.raises(ValueError):
+        opt(None, rebound='dummy')
+
+    # test weight_decay_type
+    with pytest.raises(ValueError):
+        opt(None, weight_decay_type='dummy')