From f15eebb71081b37fcd1d93348cab19a38d0934e5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laurent=20P=2E=20Ren=C3=A9=20de=20Cotret?= Date: Tue, 12 Jun 2018 13:34:35 -0400 Subject: [PATCH] Added test for register_time_shift --- skued/time_series/tests/test_time_zero.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/skued/time_series/tests/test_time_zero.py b/skued/time_series/tests/test_time_zero.py index fec55f7d..8c65aac9 100644 --- a/skued/time_series/tests/test_time_zero.py +++ b/skued/time_series/tests/test_time_zero.py @@ -50,6 +50,20 @@ def test_shift_different_lengths(self): trace2 = np.empty((8,)) 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) + class TestTimeShifts(unittest.TestCase): def test_trivial(self):