def test_uniform_interp_grid(fill, interp_kind, extrap, extrap_kind, method,
expected):
interp_grid = UniformInterpolationGrid(
fill=fill,
kind=interp_kind,
)
extrap_grid = UniformExtrapolationGrid(
interp_grid,
before=extrap,
after=extrap,
kind=extrap_kind,
)
curve = Curve([(1, 3, 5)] * 2)
curve_i = curve.interpolate(interp_grid, method=method)
curve_e = curve.interpolate(extrap_grid, method=method)
if extrap_kind == 'length':
extrap_pcount = round(extrap / curve_i.chordlen.mean()) * 2
else:
extrap_pcount = extrap * 2
extrap_arclen = curve_i.chordlen.mean() * extrap_pcount
assert extrap_grid(curve) == pytest.approx(expected)
assert curve_e.size == pytest.approx(curve_i.size + extrap_pcount)
assert curve_e.arclen == pytest.approx(curve_i.arclen + extrap_arclen)
def test_extrap_kind_point(method):
curve = Curve([(1, 3, 5, 7, 9)] * 2)
grid = UniformExtrapolationGrid(UniformInterpolationGrid(9, kind='point'),
before=3,
after=3,
kind='point')
expected = [(-2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12)] * 2
assert curve.interpolate(grid, method) == Curve(expected)
def test_extrap_kind_length(method, denom, exlen):
curve = Curve([(1, 3, 5, 7, 9)] * 2)
chordlen = curve.chordlen.mean() / denom
extraplen = chordlen * exlen
interp_grid = UniformInterpolationGrid(fill=chordlen, kind='length')
extrap_grid = UniformExtrapolationGrid(interp_grid,
before=extraplen,
after=extraplen,
kind='length')
curve_i = curve.interpolate(interp_grid, method=method)
curve_e = curve.interpolate(extrap_grid, method=method)
assert curve_e.arclen == pytest.approx(curve_i.arclen + extraplen * 2)
def test_interp(ndmin, method):
curve = Curve([(1, 3, 6, 9)] * 2, ndmin=ndmin)
expected = Curve([(1, 2, 3, 4, 5, 6, 7, 8, 9)] * 2, ndmin=ndmin)
assert np.allclose(expected.chordlen, expected.chordlen[0])
assert curve.interpolate(9, method=method) == expected