| 84 | self.allclose(mlab.detrend_mean(input.tolist(), axis=0), target) |
| 85 | |
| 86 | def test_detrend_mean_2d(self): |
| 87 | input = np.vstack([self.sig_off, |
| 88 | self.sig_base + self.sig_off]) |
| 89 | target = np.vstack([self.sig_zeros, |
| 90 | self.sig_base]) |
| 91 | self.allclose(mlab.detrend_mean(input), target) |
| 92 | self.allclose(mlab.detrend_mean(input, axis=None), target) |
| 93 | self.allclose(mlab.detrend_mean(input.T, axis=None).T, target) |
| 94 | self.allclose(mlab.detrend(input), target) |
| 95 | self.allclose(mlab.detrend(input, axis=None), target) |
| 96 | self.allclose( |
| 97 | mlab.detrend(input.T, key="constant", axis=None), target.T) |
| 98 | |
| 99 | input = np.vstack([self.sig_base, |
| 100 | self.sig_base + self.sig_off, |
| 101 | self.sig_base + self.sig_slope, |
| 102 | self.sig_base + self.sig_off + self.sig_slope]) |
| 103 | target = np.vstack([self.sig_base, |
| 104 | self.sig_base, |
| 105 | self.sig_base + self.sig_slope_mean, |
| 106 | self.sig_base + self.sig_slope_mean]) |
| 107 | self.allclose(mlab.detrend_mean(input.T, axis=0), target.T) |
| 108 | self.allclose(mlab.detrend_mean(input, axis=1), target) |
| 109 | self.allclose(mlab.detrend_mean(input, axis=-1), target) |
| 110 | self.allclose(mlab.detrend(input, key="default", axis=1), target) |
| 111 | self.allclose(mlab.detrend(input.T, key="mean", axis=0), target.T) |
| 112 | self.allclose( |
| 113 | mlab.detrend(input.T, key=mlab.detrend_mean, axis=0), target.T) |
| 114 | |
| 115 | def test_detrend_ValueError(self): |
| 116 | for signal, kwargs in [ |