def test_roll_sum_simple(self):
x = np.arange(5)
m = np.zeros(5)
rs0 = np.array([0, 1, 2, 3, 4])
rs1 = np.array([1.5, 3, 6, 9, 10.5])
self.assertTrue((mfblp_filter.roll_sum(x, m, 0) == rs0).all())
self.assertTrue((mfblp_filter.roll_sum(x, m, 1) == rs1).all())
def test_roll_sum_simple(self):
x = np.arange(5)
m = np.zeros(5)
rs0 = np.array([0, 1, 2, 3, 4])
rs1 = np.array([1.5, 3, 6, 9, 10.5])
self.assertTrue((mfblp_filter.roll_sum(x, m, 0) == rs0).all())
self.assertTrue((mfblp_filter.roll_sum(x, m, 1) == rs1).all())
def test_roll_sum_diffs(self):
x = np.array([1., 3., 0., 2., 0., 1., 4.])
m = np.array([2., 5., 1., 6., 0., 0., 4.])
rs0 = np.array([1, 2, 1, 4, 0, 1, 0])
rs1 = np.array([3, 11, 4, 16, 3, 5, 4.5])
self.assertTrue((mfblp_filter.roll_sum(x, m, 0) == rs0).all())
self.assertTrue((mfblp_filter.roll_sum(x, m, 1) == rs1).all())
def test_roll_sum_diffs(self):
x = np.array([1., 3., 0., 2., 0., 1., 4.])
m = np.array([2., 5., 1., 6., 0., 0., 4.])
rs0 = np.array([1, 2, 1, 4, 0, 1, 0])
rs1 = np.array([3, 11, 4, 16, 3, 5, 4.5])
self.assertTrue((mfblp_filter.roll_sum(x, m, 0) == rs0).all())
self.assertTrue((mfblp_filter.roll_sum(x, m, 1) == rs1).all())
