diff options
Diffstat (limited to 'test/sca/test_test.py')
| -rw-r--r-- | test/sca/test_test.py | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/test/sca/test_test.py b/test/sca/test_test.py index 898c6d6..5277c20 100644 --- a/test/sca/test_test.py +++ b/test/sca/test_test.py @@ -8,16 +8,19 @@ from pyecsca.sca import Trace, welch_ttest, student_ttest, ks_test class TTestTests(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"))) - self.d = Trace(None, b"\x00", np.array([98, 36], dtype=np.dtype("i1"))) + self.a = Trace(np.array([20, 80], dtype=np.dtype("i1")), None, b"\xff") + self.b = Trace(np.array([30, 42], dtype=np.dtype("i1")), None, b"\xff") + self.c = Trace(np.array([78, 56], dtype=np.dtype("i1")), None, b"\x00") + self.d = Trace(np.array([98, 36], dtype=np.dtype("i1")), None, b"\x00") def test_welch_ttest(self): self.assertIsNotNone(welch_ttest([self.a, self.b], [self.c, self.d])) - a = Trace(None, None, np.array([19.8, 20.4, 19.6, 17.8, 18.5, 18.9, 18.3, 18.9, 19.5, 22.0])) - b = Trace(None, None, np.array([28.2, 26.6, 20.1, 23.3, 25.2, 22.1, 17.7, 27.6, 20.6, 13.7])) - c = Trace(None, None, np.array([20.2, 21.6, 27.1, 13.3, 24.2, 20.1, 11.7, 25.6, 26.6, 21.4])) + a = Trace(np.array([19.8, 20.4, 19.6, 17.8, 18.5, 18.9, 18.3, 18.9, 19.5, 22.0]), None, + None) + b = Trace(np.array([28.2, 26.6, 20.1, 23.3, 25.2, 22.1, 17.7, 27.6, 20.6, 13.7]), None, + None) + c = Trace(np.array([20.2, 21.6, 27.1, 13.3, 24.2, 20.1, 11.7, 25.6, 26.6, 21.4]), None, + None) result = welch_ttest([a, b], [b, c]) self.assertIsNotNone(result) @@ -32,8 +35,8 @@ class KolmogorovSmirnovTests(TestCase): def test_ks_test(self): self.assertIsNone(ks_test([], [])) - a = Trace(None, b"\xff", np.array([20, 80], dtype=np.dtype("i1"))) - b = Trace(None, b"\xff", np.array([30, 42], dtype=np.dtype("i1"))) - c = Trace(None, b"\x00", np.array([78, 56], dtype=np.dtype("i1"))) - d = Trace(None, b"\x00", np.array([98, 36], dtype=np.dtype("i1"))) + a = Trace(np.array([20, 80], dtype=np.dtype("i1")), None, b"\xff") + b = Trace(np.array([30, 42], dtype=np.dtype("i1")), None, b"\xff") + c = Trace(np.array([78, 56], dtype=np.dtype("i1")), None, b"\x00") + d = Trace(np.array([98, 36], dtype=np.dtype("i1")), None, b"\x00") self.assertIsNotNone(ks_test([a, b], [c, d])) |