Migration Graph Builder bug fix (#514)

* Add test for migration builder as_dict() method * migration builder typo fix
materialsproject · Aug 26, 2022 · 5648066 · 5648066
1 parent a970b2f
commit 5648066
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/emmet-builders/emmet/builders/mobility/migration_graph.py b/emmet-builders/emmet/builders/mobility/migration_graph.py
@@ -24,7 +24,7 @@ def __init__(
         angle_tol: float = 5,
         **kwargs,
     ):
-        self.insertion_electode = insertion_electrode
+        self.insertion_electrode = insertion_electrode
         self.migration_graph = migration_graph
         self.algorithm = algorithm
         self.min_hop_distance = min_hop_distance

diff --git a/tests/emmet-builders/test_mobility.py b/tests/emmet-builders/test_mobility.py
@@ -25,3 +25,5 @@ def test_migration_graph_builder(ie_store, mg_store):
     assert mg_store.count() == 2
     assert mg_store.count({"state": "successful"}) == 2
     assert mg_store.count({"deprecated": False}) == 2
+    d = builder.as_dict()
+    assert type(d) is dict