def test_running_mean(): arr = np.array([np.arange(10), np.arange(1, 11), np.arange(2, 12)]) size = 3 result = combine_frames.running_median(arr, size) compare = np.array([np.arange(1, 9), np.arange(2, 10), np.arange(3, 11)]) assert np.array_equal(result, compare)
def test_running_mean(buffer, size): result = combine_frames.running_median(buffer, size) compare = np.array([np.arange(1, 9), np.arange(2, 10), np.arange(3, 11)]) assert np.array_equal(result, compare)