diff options
Diffstat (limited to 'test/sca/test_filter.py')
| -rw-r--r-- | test/sca/test_filter.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/test/sca/test_filter.py b/test/sca/test_filter.py index 73516d8..fda5626 100644 --- a/test/sca/test_filter.py +++ b/test/sca/test_filter.py @@ -2,10 +2,10 @@ from unittest import TestCase import numpy as np from pyecsca.sca import Trace, filter_lowpass, filter_highpass, filter_bandpass, filter_bandstop -from .utils import plot +from .utils import Plottable -class FilterTests(TestCase): +class FilterTests(Plottable): def setUp(self): self._trace = Trace( @@ -16,22 +16,22 @@ class FilterTests(TestCase): result = filter_lowpass(self._trace, 100, 20) self.assertIsNotNone(result) self.assertEqual(len(self._trace.samples), len(result.samples)) - plot(self, self._trace, result) + 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)) - plot(self, self._trace, result) + 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)) - plot(self, self._trace, result) + 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)) - plot(self, self._trace, result) + self.plot(self._trace, result) |