def test_detrend_mean_dataframe(self): x = DataFrame(np.random.randn(3, 4)) dtx = detrend_mean(x) m = dtx.mean() eps = np.finfo(float).eps assert_allclose(m.values.squeeze(), np.zeros(m.shape), atol=eps)
def test_detrend_mean(self): x = np.random.randn(3, 2) dtx = detrend_mean(x) expect = x - x.mean(0) self.assertEqual(expect.dtype, dtx.dtype) assert_array_equal(dtx, expect) assert_allclose(dtx.mean(), 0.0, atol=np.finfo(dtx.dtype).eps)