def test_set_params(self):
sk_date_column = 'ds'
sk_yhat_only = True
sk_extra_regressors = [dict(name='x', mode='multiplicative')]
prophet_kwargs = {'daily_seasonality': True}
p = SkProphet(sk_date_column, sk_yhat_only, sk_extra_regressors,
prophet_kwargs)
# Alternative params
alternative_params = {
'sk_date_column': 'date',
'sk_extra_regressors': [dict(name='z')],
}
new_params = p.get_params(deep=True)
new_params.update(alternative_params)
# Assert params
p.set_params(**new_params)
params = p.get_params(deep=False)
self.assertEqual(new_params, params)
self.assertEqual(p.sk_date_column, 'date')
self.assertEqual(len(p.sk_extra_regressors), 1)
self.assertEqual(p.sk_extra_regressors[0]['name'], 'z')
def test_get_params_shallow(self):
sk_date_column = 'ds'
sk_yhat_only = True
sk_extra_regressors = [dict(name='x', mode='multiplicative')]
prophet_kwargs = {'daily_seasonality': True}
p = SkProphet(sk_date_column, sk_yhat_only, sk_extra_regressors,
prophet_kwargs)
# Assert shallow equality
params = p.get_params(deep=False)
self.assertEqual(len(params), 4)
self.assertEqual(params['sk_date_column'], sk_date_column)
self.assertEqual(params['sk_yhat_only'], sk_yhat_only)
self.assertEqual(params['sk_extra_regressors'], sk_extra_regressors)
self.assertEqual(
params['prophet_kwargs']['daily_seasonality'],
prophet_kwargs['daily_seasonality'],
)
# Assert shallow same objects
self.assertTrue(params['sk_date_column'] is sk_date_column)
self.assertTrue(params['sk_extra_regressors'] is sk_extra_regressors)
self.assertTrue(
params['sk_extra_regressors'][0] is sk_extra_regressors[0])
self.assertTrue(params['prophet_kwargs'] is prophet_kwargs)
