1 files changed, 75 insertions, 0 deletions
diff --git a/test/test_align.py b/test/test_align.py
new file mode 100644
index 0000000..ae7baba
--- /dev/null
+++ b/test/test_align.py
@@ -0,0 +1,75 @@
+from unittest import TestCase
+
+import numpy as np
+from pyecsca import align_correlation, align_peaks, align_sad, align_dtw_scale,\
+    align_dtw, Trace, InspectorTraceSet
+from .utils import slow, plot
+
+
+class AlignTests(TestCase):
+
+    def test_align(self):
+        first_arr = np.array([10, 64, 120, 64, 10, 10, 10, 10, 10], dtype=np.dtype("i1"))
+        second_arr = np.array([10, 10, 10, 10, 50, 80, 50, 20, 10], dtype=np.dtype("i1"))
+        third_arr = np.array([70, 30, 42, 35, 28, 21, 15, 10, 5], dtype=np.dtype("i1"))
+        a = Trace(None, None, first_arr)
+        b = Trace(None, None, second_arr)
+        c = Trace(None, None, third_arr)
+        result = align_correlation(a, b, c, reference_offset=1, reference_length=3, max_offset=4, min_correlation=0.65)
+        self.assertIsNotNone(result)
+        self.assertEqual(len(result), 2)
+        np.testing.assert_equal(result[0].samples, first_arr)
+        np.testing.assert_equal(result[1].samples, np.array([10, 50, 80, 50, 20, 10, 0, 0, 0], dtype=np.dtype("i1")))
+
+    @slow
+    def test_large_align(self):
+        example = InspectorTraceSet("test/example.trs")
+        result = align_correlation(*example, reference_offset=100000, reference_length=20000, max_offset=15000)
+        self.assertIsNotNone(result)
+
+    @slow
+    def test_large_dtw_align(self):
+        example = InspectorTraceSet("test/example.trs")
+        result = align_dtw(*example[:5])
+        self.assertIsNotNone(result)
+
+    def test_peak_align(self):
+        first_arr = np.array([10, 64, 14, 120, 15, 30, 10, 15, 20, 15, 15, 10, 10], dtype=np.dtype("i1"))
+        second_arr = np.array([10, 10, 10, 10, 90, 40, 50, 20, 10, 17, 16, 10, 10], dtype=np.dtype("i1"))
+        a = Trace(None, None, first_arr)
+        b = Trace(None, None, second_arr)
+        result = align_peaks(a, b, reference_offset=2, reference_length=5, max_offset=3)
+        self.assertEqual(np.argmax(result[0].samples), np.argmax(result[1].samples))
+
+    def test_sad_align(self):
+        first_arr = np.array([10, 64, 14, 120, 15, 30, 10, 15, 20, 15, 15, 10, 10], dtype=np.dtype("i1"))
+        second_arr = np.array([10, 10, 10, 10, 90, 40, 50, 20, 10, 17, 16, 10, 10], dtype=np.dtype("i1"))
+        a = Trace(None, None, first_arr)
+        b = Trace(None, None, second_arr)
+        align_sad(a, b, reference_offset=2, reference_length=5, max_offset=3)
+
+    def test_dtw_align_scale(self):
+        first_arr = np.array([10, 64, 14, 120, 15, 30, 10, 15, 20, 15, 15, 10, 10, 8, 10, 12, 10, 13, 9], dtype=np.dtype("i1"))
+        second_arr = np.array([10, 10, 10, 10, 60, 40, 90, 20, 10, 17, 16, 10, 10, 10, 10, 10, 17, 12, 10], dtype=np.dtype("i1"))
+        third_arr = np.array([10, 30, 20, 21, 15, 8, 10, 37, 21, 77, 20, 28, 25, 10, 9, 10, 15, 9, 10], dtype=np.dtype("i1"))
+        a = Trace(None, None, first_arr)
+        b = Trace(None, None, second_arr)
+        c = Trace(None, None, third_arr)
+        result = align_dtw_scale(a, b, c)
+
+        self.assertEqual(np.argmax(result[0].samples), np.argmax(result[1].samples))
+        self.assertEqual(np.argmax(result[1].samples), np.argmax(result[2].samples))
+        plot(self, *result)
+
+    def test_dtw_align(self):
+        first_arr = np.array([10, 64, 14, 120, 15, 30, 10, 15, 20, 15, 15, 10, 10, 8, 10, 12, 10, 13, 9], dtype=np.dtype("i1"))
+        second_arr = np.array([10, 10, 10, 10, 60, 40, 90, 20, 10, 17, 16, 10, 10, 10, 10, 10, 17, 12, 10], dtype=np.dtype("i1"))
+        third_arr = np.array([10, 30, 20, 21, 15, 8, 10, 47, 21, 77, 20, 28, 25, 10, 9, 10, 15, 9, 10], dtype=np.dtype("i1"))
+        a = Trace(None, None, first_arr)
+        b = Trace(None, None, second_arr)
+        c = Trace(None, None, third_arr)
+        result = align_dtw(a, b, c)
+
+        self.assertEqual(np.argmax(result[0].samples), np.argmax(result[1].samples))
+        self.assertEqual(np.argmax(result[1].samples), np.argmax(result[2].samples))
+        plot(self, *result)