test set_params

benedekrozemberczki · Aug 11, 2023 · 45e833e · 45e833e
1 parent b327689
commit 45e833e
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/test/test_base.py b/test/test_base.py
@@ -5,4 +5,15 @@ def test_get_params():
     params = model.get_params()
     assert len(params) != 0
     assert type(params) is dict
-    assert '_embedding' not in params
+    assert '_embedding' not in params
+
+def test_set_params():
+    model = DeepWalk()
+    default_params = model.get_params()
+    params = {'dimensions': 1,
+              'seed': 123}
+    model.set_params(**params)
+    new_params = model.get_params()
+    assert new_params != default_params
+    assert new_params['dimensions'] == 1
+    assert new_params['seed'] == 123