def test_time_shift_trivial():
""" Test that the time-shift between two identical traces is zero. """
trace1 = np.sin(2 * np.pi * np.linspace(0, 10, 64))
trace2 = np.array(trace1, copy=True)
shift = register_time_shift(trace1, trace2)
assert shift == 0
trace1 = np.sin(2 * np.pi * np.linspace(0, 10, 65))
trace2 = np.array(trace1, copy=True)
shift = register_time_shift(trace1, trace2)
assert shift == 0
def test_time_shift_not1d():
""" Test that register_time_shift() raises an exception if the reference or trace are not 1D """
with pytest.raises(ValueError):
trace1 = np.empty((16, 45))
trace2 = np.empty((8, ))
register_time_shift(trace1, trace2)
with pytest.raises(ValueError):
trace1 = np.empty((16, ))
trace2 = np.empty((8, 2))
register_time_shift(trace1, trace2)
def test_shift_not1d(self):
""" Test that register_time_shift() raises an exception if the reference or trace are not 1D """
with self.subTest("Trace not 1D"):
with self.assertRaises(ValueError):
trace1 = np.empty((16, 45))
trace2 = np.empty((8, ))
register_time_shift(trace1, trace2)
with self.subTest("Reference not 1D"):
with self.assertRaises(ValueError):
trace1 = np.empty((16, ))
trace2 = np.empty((8, 2))
register_time_shift(trace1, trace2)
def test_trivial(self):
""" Test that the time-shift between two identical traces is zero. """
with self.subTest("Even length"):
trace1 = np.sin(2 * np.pi * np.linspace(0, 10, 64))
trace2 = np.array(trace1, copy=True)
shift = register_time_shift(trace1, trace2)
self.assertEqual(shift, 0)
with self.subTest("Odd length"):
trace1 = np.sin(2 * np.pi * np.linspace(0, 10, 65))
trace2 = np.array(trace1, copy=True)
shift = register_time_shift(trace1, trace2)
self.assertEqual(shift, 0)
def test_shift_with_noise(self):
""" Test measuring the time-shift between traces shifted from one another, with added 6% gaussian noise """
trace1 = np.sin(2 * np.pi * np.linspace(0, 10, 64))
trace2 = scipy_shift(trace1, 5)
trace1 = trace1[6:-6]
trace2 = trace2[6:-6]
trace1 += 0.03 * np.random.random(size=trace1.shape)
trace2 += 0.03 * np.random.random(size=trace2.shape)
shift = register_time_shift(trace1, trace2)
self.assertEqual(shift, -5)
def test_shift_different_lengths(self):
""" Test that register_time_shift() raises an exception if the reference and trace do not have the same shape """
with self.assertRaises(ValueError):
trace1 = np.empty((16, ))
trace2 = np.empty((8, ))
register_time_shift(trace1, trace2)
def test_shift_no_noise(self):
""" Test measuring the time-shift between traces shifted from one another, without added noise """
trace1 = np.sin(2 * np.pi * np.linspace(0, 10, 64))
trace2 = np.roll(trace1, 5)
shift = register_time_shift(trace1, trace2)
self.assertEqual(shift, -5)
