add iter test

pynamodb · Oct 7, 2017 · 5f21e27 · 5f21e27
1 parent b18bd55
commit 5f21e27
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pynamodb/tests/test_attributes.py b/pynamodb/tests/test_attributes.py
@@ -644,6 +644,18 @@ def test_raw_map_access(self):
         for k, v in six.iteritems(raw):
             assert attr[k] == v
 
+    def test_raw_map_iter(self):
+        raw = {
+            "foo": "bar",
+            "num": 3,
+            "nested": {
+                "nestedfoo": "nestedbar"
+            }
+        }
+        attr = MapAttribute(**raw)
+
+        assert list(iter(raw)) == list(iter(attr))
+
     def test_raw_map_json_serialize(self):
         raw = {
             "foo": "bar",