update test to use set attribute

pynamodb · Sep 5, 2020 · 896a198 · 896a198
1 parent a351efe
commit 896a198
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_attributes.py b/tests/test_attributes.py
@@ -54,7 +54,7 @@ class CustomAttrMap(MapAttribute):
 
 class DefaultsMap(MapAttribute):
     map_field = MapAttribute(default={})
-    string_field = UnicodeAttribute(null=True)
+    string_set_field = UnicodeSetAttribute(null=True)
 
 
 class TestAttributeDescriptor:
@@ -624,7 +624,7 @@ def test_null_attribute_subclassed_map(self):
 
     def test_null_attribute_map_after_serialization(self):
         null_attribute = {
-            'string_field': '',
+            'string_set_field': {},
         }
         attr = DefaultsMap()
         serialized = attr.serialize(null_attribute)