diff options
| author | J08nY | 2021-04-10 17:50:05 +0200 |
|---|---|---|
| committer | J08nY | 2021-04-10 17:50:05 +0200 |
| commit | b76ec0890e4cf997ce5a0b4494722931094683f7 (patch) | |
| tree | 2dc049de8a79adc7b3b23d746ef4f28d58e33fbc /test/sca/test_match.py | |
| parent | a7ad11f7cd917be55dbd036a516fefda4d19dd4a (diff) | |
| download | pyecsca-b76ec0890e4cf997ce5a0b4494722931094683f7.tar.gz pyecsca-b76ec0890e4cf997ce5a0b4494722931094683f7.tar.zst pyecsca-b76ec0890e4cf997ce5a0b4494722931094683f7.zip | |
Use black.
Diffstat (limited to 'test/sca/test_match.py')
| -rw-r--r-- | test/sca/test_match.py | 66 |
1 files changed, 56 insertions, 10 deletions
diff --git a/test/sca/test_match.py b/test/sca/test_match.py index cd0b780..9ba81a3 100644 --- a/test/sca/test_match.py +++ b/test/sca/test_match.py @@ -7,22 +7,68 @@ from .utils import Plottable class MatchingTests(Plottable): - def test_simple_match(self): - pattern = Trace(np.array([1, 15, 12, -10, 0, 13, 17, -1, 0], dtype=np.dtype("i1")), None) - base = Trace(np.array( + pattern = Trace( + np.array([1, 15, 12, -10, 0, 13, 17, -1, 0], dtype=np.dtype("i1")), None + ) + base = Trace( + np.array( [0, 1, 3, 1, 2, -2, -3, 1, 15, 12, -10, 0, 13, 17, -1, 0, 3, 1], - dtype=np.dtype("i1")), None) + dtype=np.dtype("i1"), + ), + None, + ) filtered = match_part(base, 7, 9) self.assertListEqual(filtered, [7]) self.plot(base=base, pattern=pad(pattern, (filtered[0], 0))) def test_multiple_match(self): - pattern = Trace(np.array([1, 15, 12, -10, 0, 13, 17, -1, 0], dtype=np.dtype("i1")), None) - base = Trace(np.array( - [0, 1, 3, 1, 2, -2, -3, 1, 18, 10, -5, 0, 13, 17, -1, 0, 3, 1, 2, 5, 13, 8, -8, 1, - 11, 15, 0, 1, 5, 2, 4], - dtype=np.dtype("i1")), None) + pattern = Trace( + np.array([1, 15, 12, -10, 0, 13, 17, -1, 0], dtype=np.dtype("i1")), None + ) + base = Trace( + np.array( + [ + 0, + 1, + 3, + 1, + 2, + -2, + -3, + 1, + 18, + 10, + -5, + 0, + 13, + 17, + -1, + 0, + 3, + 1, + 2, + 5, + 13, + 8, + -8, + 1, + 11, + 15, + 0, + 1, + 5, + 2, + 4, + ], + dtype=np.dtype("i1"), + ), + None, + ) filtered = match_pattern(base, pattern, 0.9) self.assertListEqual(filtered, [7, 19]) - self.plot(base=base, pattern1=pad(pattern, (filtered[0], 0)), pattern2=pad(pattern, (filtered[1], 0))) + self.plot( + base=base, + pattern1=pad(pattern, (filtered[0], 0)), + pattern2=pad(pattern, (filtered[1], 0)), + ) |