test_utilities.py (prophet-0.7) | : | test_utilities.py (prophet-1.0) | ||
---|---|---|---|---|
skipping to change at line 16 | skipping to change at line 16 | |||
from __future__ import absolute_import | from __future__ import absolute_import | |||
from __future__ import division | from __future__ import division | |||
from __future__ import print_function | from __future__ import print_function | |||
from __future__ import unicode_literals | from __future__ import unicode_literals | |||
import os | import os | |||
from unittest import TestCase | from unittest import TestCase | |||
import numpy as np | import numpy as np | |||
import pandas as pd | import pandas as pd | |||
from fbprophet import Prophet | from prophet import Prophet | |||
from fbprophet.utilities import regressor_coefficients | from prophet.utilities import regressor_coefficients | |||
DATA = pd.read_csv( | DATA = pd.read_csv( | |||
os.path.join(os.path.dirname(__file__), 'data.csv'), | os.path.join(os.path.dirname(__file__), 'data.csv'), | |||
parse_dates=['ds'], | parse_dates=['ds'], | |||
) | ) | |||
class TestUtilities(TestCase): | class TestUtilities(TestCase): | |||
def test_regressor_coefficients(self): | def test_regressor_coefficients(self): | |||
m = Prophet() | m = Prophet() | |||
N = DATA.shape[0] | N = DATA.shape[0] | |||
End of changes. 1 change blocks. | ||||
2 lines changed or deleted | 2 lines changed or added |