1 files changed, 35 insertions, 0 deletions
diff --git a/test/test_combine.py b/test/test_combine.py
new file mode 100644
index 0000000..ca23197
--- /dev/null
+++ b/test/test_combine.py
@@ -0,0 +1,35 @@
+from unittest import TestCase
+
+import numpy as np
+from pyecsca import Trace, CombinedTrace, average, conditional_average, standard_deviation
+
+
+class CombineTests(TestCase):
+
+    def setUp(self):
+        self.a = Trace(None, b"\xff", np.array([20, 80], dtype=np.dtype("i1")))
+        self.b = Trace(None, b"\xff", np.array([30, 42], dtype=np.dtype("i1")))
+        self.c = Trace(None, b"\x00", np.array([78, 56], dtype=np.dtype("i1")))
+
+    def test_average(self):
+        self.assertIsNone(average())
+        result = average(self.a, self.b)
+        self.assertIsNotNone(result)
+        self.assertIsInstance(result, CombinedTrace)
+        self.assertEqual(len(result.samples), 2)
+        self.assertEqual(result.samples[0], 25)
+        self.assertEqual(result.samples[1], 61)
+
+    def test_conditional_average(self):
+        result = conditional_average(self.a, self.b, self.c,
+                                     condition=lambda trace: trace.data[0] == 0xff)
+        self.assertIsInstance(result, CombinedTrace)
+        self.assertEqual(len(result.samples), 2)
+        self.assertEqual(result.samples[0], 25)
+        self.assertEqual(result.samples[1], 61)
+
+    def test_standard_deviation(self):
+        self.assertIsNone(standard_deviation())
+        result = standard_deviation(self.a, self.b)
+        self.assertIsInstance(result, CombinedTrace)
+        self.assertEqual(len(result.samples), 2)