test: add more test case

01Joseph-Hwang10 · Mar 6, 2024 · 0a8b64d · 0a8b64d
1 parent 6ac052e
commit 0a8b64d
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_dotty_custom_types.py b/tests/test_dotty_custom_types.py
@@ -123,5 +123,20 @@ def test_dotty_del_custom_types(dot: Dotty):
     assert dot.get("h.name") is None
 
 
+def test_dotty_views_custom_types(dot: Dotty):
+    assert list(dot) == ["a", "g", "h"]
+    assert list(dot.keys()) == ["a", "g", "h"]
+    assert list(dot.values()) == [
+        dot["a"],
+        dot["g"],
+        dot["h"],
+    ]
+    assert list(dot.items()) == [
+        ("a", dot["a"]),
+        ("g", dot["g"]),
+        ("h", dot["h"]),
+    ]
+
+
 def test_dotty_to_dict(dot: Dotty, dot_serialized):
     assert dot.to_dict() == dot_serialized