def test_shape_power_law(self): exponent = -2.5 f = Shape._power_law(exponent) k = np.arange(1, 10, dtype=float) np.testing.assert_almost_equal(f(k), k**exponent)
def test_shape_power_law_amp(self): exponent = -2.5 amp = 10 f = Shape._power_law(exponent, amp=amp) k = np.arange(1, 10, dtype=float) np.testing.assert_almost_equal(f(k), amp * k**exponent)