| 61 | assert mlab.detrend_none(sig) == sig |
| 62 | |
| 63 | def test_detrend_mean(self): |
| 64 | for sig in [0., 5.5]: # 0D. |
| 65 | assert mlab.detrend_mean(sig) == 0. |
| 66 | assert mlab.detrend(sig, key="mean") == 0. |
| 67 | assert mlab.detrend(sig, key=mlab.detrend_mean) == 0. |
| 68 | # 1D. |
| 69 | self.allclose(mlab.detrend_mean(self.sig_zeros), self.sig_zeros) |
| 70 | self.allclose(mlab.detrend_mean(self.sig_base), self.sig_base) |
| 71 | self.allclose(mlab.detrend_mean(self.sig_base + self.sig_off), |
| 72 | self.sig_base) |
| 73 | self.allclose(mlab.detrend_mean(self.sig_base + self.sig_slope), |
| 74 | self.sig_base + self.sig_slope_mean) |
| 75 | self.allclose( |
| 76 | mlab.detrend_mean(self.sig_base + self.sig_slope + self.sig_off), |
| 77 | self.sig_base + self.sig_slope_mean) |
| 78 | |
| 79 | def test_detrend_mean_1d_base_slope_off_list_andor_axis0(self): |
| 80 | input = self.sig_base + self.sig_slope + self.sig_off |