def test_shrink_args(self):
traces = [
CallTrace(tie_helper, {
'a': str,
'b': int
}),
CallTrace(tie_helper, {
'a': str,
'b': NoneType
}),
]
assert shrink_traced_types(traces) == ({
'a': str,
'b': Optional[int]
}, None, None)
def test_shrink_yield(self):
traces = [
CallTrace(tie_helper, {}, yield_type=int),
CallTrace(tie_helper, {}, yield_type=str),
]
assert shrink_traced_types(traces) == ({}, None, Union[int, str])
def test_shrink_return(self):
traces = [
CallTrace(tie_helper, {}, NoneType),
CallTrace(tie_helper, {}, str),
]
assert shrink_traced_types(traces) == ({}, Optional[str], None)
