def test_shapes_gufunc_arg_shapes(parsed_sig_and_size, data):
parsed_sig, min_side, max_side = parsed_sig_and_size
# This private function assumes already preprocessed sizes to default dict
min_side = gu._int_or_dict(min_side, 0)
max_side = gu._int_or_dict(max_side, gu.DEFAULT_MAX_SIDE)
S = gu._gufunc_arg_shapes(parsed_sig[0], min_side=min_side, max_side=max_side)
shapes = data.draw(S)
validate_shapes(shapes, parsed_sig[0], min_side, max_side)
def test_ddict_int_or_dict(D, default_val, default_val2):
DD = defaultdict(lambda: default_val, D)
DD2 = gu._int_or_dict(DD, default_val2)
# just pass thru
assert DD is DD2
# default_val2 is ignored
assert DD2.default_factory() == default_val
def test_int_int_or_dict(default_val, default_val2):
DD = gu._int_or_dict(default_val, default_val2)
assert len(DD) == 0
assert DD["---"] == default_val
def test_dict_int_or_dict(D, default_val):
DD = gu._int_or_dict(D, default_val)
assert DD == D
assert DD["---"] == default_val
