Add deprecated check for MigratrionGraphBuilder test

materialsproject · Aug 1, 2022 · 98b71e6 · 98b71e6
1 parent 695995f
commit 98b71e6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/emmet-builders/test_mobility.py b/tests/emmet-builders/test_mobility.py
@@ -6,7 +6,9 @@
 
 @pytest.fixture(scope="session")
 def ie_store(test_dir):
-    return JSONStore(test_dir / "mobility/builder_migration_graph_set.json", key="battery_id")
+    return JSONStore(
+        test_dir / "mobility/builder_migration_graph_set.json", key="battery_id"
+    )
 
 
 @pytest.fixture
@@ -22,3 +24,4 @@ def test_migration_graph_builder(ie_store, mg_store):
     builder.run()
     assert mg_store.count() == 2
     assert mg_store.count({"state": "successful"}) == 2
+    assert mg_store.count({"deprecated": False}) == 2