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)