def test_consecutive_index_all_equal(self):
"""Test for helper function to find consecutive index with identical numbers."""
vec = np.array([1, 1, 1, 1, 1, 1])
ref_inds = np.array([True, False, False, False, False, True], dtype=bool)
inds = waveform._find_consecutive_index(vec, resolution=1e-6)
np.testing.assert_array_equal(inds, ref_inds)
def test_consecutive_index_tiny_diff(self):
"""Test for helper function to find consecutive index with vector with tiny change."""
eps = 1e-10
vec = np.array([0.5, 0.5+eps, 0.5-eps, 0.5+eps, 0.5-eps, 0.5])
ref_inds = np.array([True, False, False, False, False, True], dtype=bool)
inds = waveform._find_consecutive_index(vec, resolution=1e-6)
np.testing.assert_array_equal(inds, ref_inds)