def test_upload_risk_model_calendar(mocker):
calendar = RiskModelCalendar.from_dict(
{"businessDates": ["2020-01-01", "2020-11-01"]})
# mock GsSession
mocker.patch.object(GsSession.__class__,
'default_value',
return_value=GsSession.get(Environment.QA, 'client_id',
'secret'))
mocker.patch.object(GsSession.current, '_put', return_value=calendar)
# run test
response = GsRiskModelApi.upload_risk_model_calendar(
'WW_TEST_MODEL', calendar)
GsSession.current._put.assert_called_with(
'/risk/models/{id}/calendar'.format(id='WW_TEST_MODEL'),
calendar,
cls=RiskModelCalendar)
assert response == calendar
def upload_calendar(self, calendar: RiskModelCalendar):
""" Upload risk model calendar to existing risk model """
return GsRiskModelApi.upload_risk_model_calendar(
self.model.id, calendar)
def upload_calendar(self, calendar: RiskModelCalendar):
""" Upload risk model calendar to existing risk model
:param calendar: RiskModelCalendar containing list of dates where model data is expected"""
return GsRiskModelApi.upload_risk_model_calendar(self.id, calendar)
