Improve tagged_union coverage

python-attrs · Dec 2, 2024 · 2eedc81 · 2eedc81
1 parent 2f464b9
commit 2eedc81
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/strategies/test_tagged_unions.py b/tests/strategies/test_tagged_unions.py
@@ -161,7 +161,10 @@ class B:
     configure_tagged_union(Union[A, B], c, default=A)
 
     data = c.unstructure(A(), Union[A, B])
-    c.structure(data, Union[A, B])
+    assert c.structure(data, Union[A, B]) == A()
+
+    data.pop("_type")
+    assert c.structure(data, Union[A, B]) == A()
 
 
 def test_nested_sequence_union():