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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#!/usr/bin/env python
# Time-stamp: <2024-05-14 13:55:10 Tao Liu>
"""Module Description: Test functions for Signal.pyx
This code is free software; you can redistribute it and/or modify it
under the terms of the BSD License (see the file LICENSE included with
the distribution).
"""
import unittest
import pytest
from math import log10
import numpy as np
from MACS3.Signal.SignalProcessing import maxima, savitzky_golay, savitzky_golay_order2_deriv1
# ------------------------------------
# Main function
# ------------------------------------
class Test_maxima(unittest.TestCase):
def setUp(self):
self.signal = np.array( [ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 5, 5, 5,
5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
5, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 7, 7, 7,
7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 6, 6, 6, 7, 7, 7, 7, 7, 7, 7, 7,
7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 7, 7, 7, 7, 7, 7, 7, 7, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
4, 4, 4, 4, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
3, 3, 3, 3, 3, 3 ], dtype = "float32" )
self.windowsize = 253
self.summit = 161 # this is based on 1-deriv smoothed data
self.smoothed162 = -2.98155597e-18 # the value is based on python3.7+numpy1.17.4; from smoothing func with np.convolve
# this test uses only numpy functions. we aim to capture strange behavior under specific py+numpy
@pytest.mark.skip( reason="it fails under some combinations of py+np with unknown reason." )
def test_implement_smooth_here ( self ):
signal = self.signal
window_size = self.windowsize
half_window = (window_size - 1) // 2
# precompute coefficients
b = np.array([[1, k, k**2] for k in range(-half_window, half_window+1)], dtype='int64')
m = np.linalg.pinv(b)[1]
# pad the signal at the extremes with
# values taken from the signal itself
firstvals = signal[0] - np.abs(signal[1:half_window+1][::-1] - signal[0])
lastvals = signal[-1] + np.abs(signal[-half_window-1:-1][::-1] - signal[-1])
signal = np.concatenate((firstvals, signal, lastvals))
ret = np.convolve( m[::-1], signal.astype("float64"), mode='valid').astype("float32") # convolve function seems to have odd ret with spec py+numpy
p = ret[162]
print ("calculated step by step:\n", p)
print ("expected:\n", self.smoothed162)
self.assertAlmostEqual( p, self.smoothed162, places = 16 )
self.assertEqual( np.sign(p), np.sign(self.smoothed162) )
def test_maxima(self):
expect = self.summit
result = maxima( self.signal, self.windowsize )[0]
self.assertEqual( result, expect, msg=f"Not equal: result: {result}, expected: {expect}" )
def assertEqual_nparray1d ( self, a, b, places = 7 ):
self.assertEqual( a.shape[0], b.shape[0] )
l = a.shape[0]
for i in range( l ):
self.assertAlmostEqual( a[i], b[i], places = places, msg=f"Not equal at {i} {a[i]} {b[i]}" )
|
