def _assert_steps_and_alphas(
self,
n,
expected_step_sizes,
expected_step_sizes_trapezoid,
expected_left,
expected_right,
expected_middle,
expected_trapezoid,
):
step_sizes_left, alphas_left = riemann_builders(Riemann.left)
step_sizes_right, alphas_right = riemann_builders(Riemann.right)
step_sizes_middle, alphas_middle = riemann_builders(Riemann.middle)
step_sizes_trapezoid, alphas_trapezoid = riemann_builders(Riemann.trapezoid)
assertArraysAlmostEqual(expected_step_sizes, step_sizes_left(n))
assertArraysAlmostEqual(expected_step_sizes, step_sizes_right(n))
assertArraysAlmostEqual(expected_step_sizes, step_sizes_middle(n))
assertArraysAlmostEqual(expected_step_sizes_trapezoid, step_sizes_trapezoid(n))
assertArraysAlmostEqual(expected_left, alphas_left(n))
assertArraysAlmostEqual(expected_right, alphas_right(n))
assertArraysAlmostEqual(expected_middle, alphas_middle(n))
assertArraysAlmostEqual(expected_trapezoid, alphas_trapezoid(n))
def test_riemann_0(self):
with self.assertRaises(AssertionError):
step_sizes, alphas = riemann_builders()
step_sizes(0)
alphas(0)
def _assert_steps_and_alphas(
self,
n,
expected_step_sizes,
expected_step_sizes_trapezoid,
expected_left,
expected_right,
expected_middle,
expected_trapezoid,
):
step_sizes_left, alphas_left = riemann_builders(Riemann.left)
step_sizes_right, alphas_right = riemann_builders(Riemann.right)
step_sizes_middle, alphas_middle = riemann_builders(Riemann.middle)
step_sizes_trapezoid, alphas_trapezoid = riemann_builders(
Riemann.trapezoid)
assertTensorAlmostEqual(
self,
torch.tensor(expected_step_sizes),
step_sizes_left(n),
delta=0.05,
mode="max",
)
assertTensorAlmostEqual(
self,
torch.tensor(expected_step_sizes),
step_sizes_right(n),
delta=0.05,
mode="max",
)
assertTensorAlmostEqual(
self,
torch.tensor(expected_step_sizes),
step_sizes_middle(n),
delta=0.05,
mode="max",
)
assertTensorAlmostEqual(
self,
torch.tensor(expected_step_sizes_trapezoid),
step_sizes_trapezoid(n),
delta=0.05,
mode="max",
)
assertTensorAlmostEqual(self,
torch.tensor(expected_left),
alphas_left(n),
delta=0.05,
mode="max")
assertTensorAlmostEqual(self,
torch.tensor(expected_right),
alphas_right(n),
delta=0.05,
mode="max")
assertTensorAlmostEqual(
self,
torch.tensor(expected_middle),
alphas_middle(n),
delta=0.05,
mode="max",
)
assertTensorAlmostEqual(
self,
torch.tensor(expected_trapezoid),
alphas_trapezoid(n),
delta=0.05,
mode="max",
)
