def test_transform_err(): with pytest.raises(ValueError): bc = BoxCox(on_err='raise') bc.fit_transform([1, 1, 1]) with pytest.raises(FloatingPointError): bc = BoxCox(on_err='raise') bc.fit_transform([1e20, 1, 1e20])
def test_transform_err(): bc = BoxCox(on_err='raise') try: bc.fit_transform([1, 1, 1]) except FloatingPointError: assert True else: assert False, 'should got FloatingPointError'
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_1(data): bc = BoxCox() trans = bc.fit_transform(data[0]) assert np.all(trans == data[2]) assert trans.shape == data[2].shape inverse = bc.inverse_transform(trans) assert np.allclose(inverse, data[0]) assert inverse.shape == data[0].shape
def test_transform_4x1_2(data): from scipy.special import boxcox as bc_ shift = 1e-5 bc = BoxCox(shift=shift) _data = data[0] - 2. trans = bc.fit_transform(_data) tmp = bc_(_data + (shift - _data.min()), bc.lambda_[0]) assert np.all(trans == tmp) inverse = bc.inverse_transform(trans) assert np.allclose(inverse, _data)
def test_transform_4x1_ravel_1(data): bc = BoxCox() trans = bc.fit_transform(data[0].ravel()) assert np.all(trans == data[2].ravel()) inverse = bc.inverse_transform(trans) assert np.allclose(inverse, data[0].ravel())
def test_transform_err_4x4_3(data): bc = BoxCox(on_err='log') trans = bc.fit_transform(data[5]) assert np.all(trans == data[7]) inverse = bc.inverse_transform(trans) assert np.all(inverse == data[5])
def test_transform_err_4x4_2(data): bc = BoxCox(on_err='nan') trans = bc.fit_transform(data[5]) assert np.all(np.isnan(trans))
def test_transform_err_4x4(data): bc = BoxCox() trans = bc.fit_transform(data[5]) assert np.all(trans == data[5]) inverse = bc.inverse_transform(trans) assert np.all(inverse == data[5])
def test_transform_4x4_2(data): bc = BoxCox() trans = bc.fit_transform(data[1]) assert np.all(trans == data[3]) inverse = bc.inverse_transform(trans) assert np.allclose(inverse, data[1])