1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
import numpy as np
from pyecsca.sca import (
Trace,
filter_lowpass,
filter_highpass,
filter_bandpass,
filter_bandstop,
)
from .utils import Plottable
class FilterTests(Plottable):
def setUp(self):
self._trace = Trace(
np.array(
[5, 12, 15, 13, 15, 11, 7, 2, -4, -8, -10, -8, -13, -9, -11, -8, -5],
dtype=np.dtype("i1"),
),
None,
)
def test_lowpass(self):
result = filter_lowpass(self._trace, 100, 20)
self.assertIsNotNone(result)
self.assertEqual(len(self._trace.samples), len(result.samples))
self.plot(self._trace, result)
def test_highpass(self):
result = filter_highpass(self._trace, 128, 20)
self.assertIsNotNone(result)
self.assertEqual(len(self._trace.samples), len(result.samples))
self.plot(self._trace, result)
def test_bandpass(self):
result = filter_bandpass(self._trace, 128, 20, 60)
self.assertIsNotNone(result)
self.assertEqual(len(self._trace.samples), len(result.samples))
self.plot(self._trace, result)
def test_bandstop(self):
result = filter_bandstop(self._trace, 128, 20, 60)
self.assertIsNotNone(result)
self.assertEqual(len(self._trace.samples), len(result.samples))
self.plot(self._trace, result)
|
