Test type casting of extra regressors

This commit is contained in:
Ben Letham 2018-12-03 15:43:13 -08:00
parent 13d96cff8f
commit b76ace9570

View file

@ -589,7 +589,7 @@ class TestProphet(TestCase):
) )
m.add_regressor('binary_feature2', standardize=True) m.add_regressor('binary_feature2', standardize=True)
df = DATA.copy() df = DATA.copy()
df['binary_feature'] = [0] * 255 + [1] * 255 df['binary_feature'] = ['0'] * 255 + ['1'] * 255
df['numeric_feature'] = range(510) df['numeric_feature'] = range(510)
df['numeric_feature2'] = range(510) df['numeric_feature2'] = range(510)
with self.assertRaises(ValueError): with self.assertRaises(ValueError):