for s2 in SHAPES:
for axis in range(-len(s2), len(s2)):
if ip != CubicSpline:
yield check_shape, ip, s1, s2, None, axis
else:
for bc in ["natural", "clamped"]:
extra = {"bc_type": bc}
yield check_shape, ip, s1, s2, None, axis, extra
After Change
for s2 in SHAPES:
for axis in range(-len(s2), len(s2)):
if ip != CubicSpline:
check_shape(ip, s1, s2, None, axis)
else:
for bc in ["natural", "clamped"]:
extra = {"bc_type": bc}
check_shape(ip, s1, s2, None, axis, extra)