Default set to None

anna-money · Dec 11, 2023 · 6655197 · 6655197
1 parent 5c3fcdb
commit 6655197
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_serialization.py b/tests/test_serialization.py
@@ -156,14 +156,19 @@ class SimpleTypesContainers:
         optional_enum_field="even",
     )
 
-    raw_no_defaults = {k: v for k, v in raw.items() if not k.endswith("default") or not k.endswith("default_factory")}
+    raw_no_defaults = {k: v for k, v in raw.items() if not k.endswith("default") and not k.endswith("default_factory")}
+    raw_defaults_set_to_none = {
+        k: (v if not k.endswith("default") and not k.endswith("default_factory") else None) for k, v in raw.items()
+    }
 
     loaded = mr.load(SimpleTypesContainers, raw)
     loaded_no_defaults = mr.load(SimpleTypesContainers, raw_no_defaults)
+    loaded_defaults_set_to_none = mr.load(SimpleTypesContainers, raw_defaults_set_to_none)
     dumped = mr.dump(loaded)
 
     assert (
         loaded_no_defaults
+        == loaded_defaults_set_to_none
         == loaded
         == SimpleTypesContainers(
             any_field={},