fix unittest error in customAttrMap for UnicodeSetAttribute

supercoderhawk · supercoderhawk · commit 8a03b6c8c060 · 2019-05-18T15:19:39.000+08:00
diff --git a/pynamodb/tests/test_attributes.py b/pynamodb/tests/test_attributes.py
@@ -642,9 +642,12 @@ def test_map_overridden_attrs_accessors(self):
     def test_map_overridden_attrs_serialize(self):
         attribute = {
             'overridden_number_attr': 10,
-            'overridden_unicode_attr': "Hello"
+            'overridden_unicode_attr': "Hello",
+            'overridden_unicode_set_attr': [six.u('Hello'), six.u('world')]
         }
-        expected = {'number_attr': {'N': '10'}, 'unicode_attr': {'S': six.u('Hello')}}
+        expected = {'number_attr': {'N': '10'},
+                    'unicode_attr': {'S': six.u('Hello')},
+                    'unicode_set_attr': {'SS': [six.u('Hello'), six.u('world')]}}
         assert CustomAttrMap().serialize(attribute) == expected
 
     def test_additional_attrs_deserialize(self):
