def test_manager_change_params(monkeypatch):
fake_params = copy.deepcopy(model.DividendsModel.PARAMS)
fake_params['data']['lags'] += 1
monkeypatch.setattr(model.DividendsModel, 'PARAMS', fake_params)
positions = tuple(['PRTK', 'MVID', 'CHMF', 'MTSS', 'PMSBP'])
date = pd.Timestamp('2018-09-06')
data = manager.DividendsMLDataManager(positions, date)
assert data.value.params['data']['lags'] == fake_params['data']['lags']
assert data.value.params == model.DividendsModel(positions, date).params
def test_manager_first_time():
positions = tuple(['MSTT', 'MVID', 'CHMF', 'MTSS', 'PMSBP'])
date = pd.Timestamp('2018-09-06')
data = manager.DividendsMLDataManager(positions, date)
assert data.value.params == model.DividendsModel(positions, date).params
def test_download_update():
positions = tuple(['PRTK', 'MVID', 'CHMF', 'MTSS', 'PMSBP'])
date = pd.Timestamp('2018-09-06')
with pytest.raises(NotImplementedError):
manager.DividendsMLDataManager(positions, date).download_update()