def test_transform_4x4_1(data): bc = BoxCox() bc.fit_transform(data[1]) trans = bc.transform(data[1][:2]) assert trans.shape == data[3][:2].shape assert np.all(trans == data[3][:2]) inverse = bc.inverse_transform(trans) assert inverse.shape == data[1][:2].shape assert np.allclose(inverse, data[1][:2])
def test_transform_4x1_ravel_2(data): bc = BoxCox() bc.fit(data[0].ravel()) trans = bc.transform((data[0].ravel())[:2]) assert trans.shape == (data[2].ravel())[:2].shape assert np.all(trans == (data[2].ravel())[:2]) inverse = bc.inverse_transform(trans) assert inverse.shape == (data[0].ravel())[:2].shape assert np.allclose(inverse, (data[0].ravel())[:2])