def test_as_np_vector_dataframe_2d(self):
p = SkProphet('date', True, [], {})
y = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]})
_y = p._as_np_vector(y)
self.assertIsInstance(_y, np.ndarray)
self.assertEqual(len(_y.shape), 1)
self.assertEqual(_y.tolist(), [1, 2, 3])
def test_as_np_vector_ndarray_2d(self):
p = SkProphet('date', True, [], {})
y = np.array([[1, 4], [2, 5], [3, 6]])
_y = p._as_np_vector(y)
self.assertIsInstance(_y, np.ndarray)
self.assertEqual(len(_y.shape), 1)
self.assertEqual(_y.tolist(), [1, 2, 3])
def test_as_np_vector_series(self):
p = SkProphet('date', True, [], {})
y = pd.Series([1, 2, 3])
_y = p._as_np_vector(y)
self.assertIsInstance(_y, np.ndarray)
self.assertEqual(len(_y.shape), 1)
self.assertEqual(_y.tolist(), [1, 2, 3])
