Merge pull request #2 from tomlincr/develop

add tests for base class Estimator.get_params()
tomlincr · Aug 21, 2022 · 4a488e0 · 4a488e0
2 parents ffd6487 + b862790
commit 4a488e0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/test_base.py b/test/test_base.py
@@ -0,0 +1,8 @@
+from karateclub import DeepWalk
+
+def test_get_params():
+    model = DeepWalk()
+    params = model.get_params()
+    assert len(params) != 0
+    assert type(params) is dict
+    assert '_embedding' not in params