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)