Added test for register_time_shift

LaurentRDC · Jun 12, 2018 · f15eebb · f15eebb
1 parent 62beb54
commit f15eebb
Showing 1 changed file with 14 additions and 0 deletions.
diff --git 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):