Expand source code
@pytest.mark.parametrize(
"r",
[
np.arange(6) * u.au,
np.arange(6) * u.cm * 1.496e13,
]
)
@pytest.mark.parametrize(
"input, expected",
[
[[[1 * u.au, 100 * u.au, 2, 1]], np.arange(6) ** 2 * u.dimensionless_unscaled],
[[[1 * u.au, 100 * u.au, 2, 1], [1 * u.au, 100 * u.au, 2, 3]],
4 * np.arange(6) ** 2 * u.dimensionless_unscaled],
[[[1 * u.au, 100 * u.au, 2, 1], [1 * u.au, 2.1 * u.au, 1, 3]],
np.array([0, 4, 10, 9, 16, 25]) * u.dimensionless_unscaled],
]
)
def test_piecewisepowerlaw(input, r, expected):
assert u.allclose(PieceWisePowerLaw(input)(r), expected, rtol=1e-4)