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())