def test_pad(self):
result = pad(self._trace, 2)
self.assertIsNotNone(result)
np.testing.assert_equal(
result.samples,
np.array([0, 0, 10, 20, 30, 40, 50, 0, 0], dtype=np.dtype("i1")))
result = pad(self._trace, (1, 3))
self.assertIsNotNone(result)
np.testing.assert_equal(
result.samples,
np.array([0, 10, 20, 30, 40, 50, 0, 0, 0], dtype=np.dtype("i1")))
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,
)
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)),
)
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([
0, 1, 3, 1, 2, -2, -3, 1, 15, 12, -10, 0, 13, 17, -1, 0, 3, 1
],
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)))