diff --git a/tests/strategies/test_tagged_unions.py b/tests/strategies/test_tagged_unions.py index abd38fef..ba5ae49b 100644 --- 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():