add test_mongo_dict

src/monty/collections.py

@@ -151,11 +151,11 @@ class MongoDict:
     a nested dict interactively (e.g. documents extracted from a MongoDB
     database).
 
-    >>> m = MongoDict({'a': {'b': 1}, 'x': 2})
-    >>> assert m.a.b == 1 and m.x == 2
-    >>> assert "a" in m and "b" in m.a
-    >>> m["a"]
-    {'b': 1}
+    >>> m_dct = MongoDict({"a": {"b": 1}, "x": 2})
+    >>> assert m_dct.a.b == 1 and m_dct.x == 2
+    >>> assert "a" in m_dct and "b" in m_dct.a
+    >>> m_dct["a"]
+    {"b": 1}
 
     Notes:
         Cannot inherit from ABC collections.Mapping because otherwise

tests/test_collections.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
 
 import os
-from collections import namedtuple
 
 import pytest
 
 from monty.collections import (
     AttrDict,
     FrozenAttrDict,
+    MongoDict,
     Namespace,
     dict2namedtuple,
     frozendict,
@@ -54,6 +54,8 @@ def test_attr_dict():
     dct = AttrDict(foo=1, bar=2)
     assert dct.bar == 2
     assert dct["foo"] is dct.foo
+
+    # Test accessing values
     dct.bar = "hello"
     assert dct["bar"] == "hello"
 
@@ -77,7 +79,12 @@ def test_frozen_attrdict():
 
 
 def test_mongo_dict():
-    """TODO: add test"""
+    m_dct = MongoDict({"a": {"b": 1}, "x": 2})
+    assert m_dct.a.b == 1
+    assert m_dct.x == 2
+    assert "a" in m_dct
+    assert "b" in m_dct.a
+    assert m_dct["a"] == {"b": 1}
 
 
 def test_dict2namedtuple():