示例#1
0
def test_update_risk_model_factor(mocker):
    factor = RiskModelFactor.from_dict({
        "identifier": "factor",
        "type": "Factor"
    })

    # 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=factor)

    # run test
    response = GsRiskModelApi.update_risk_model_factor(model_id='id',
                                                       factor_id='factor',
                                                       factor=factor)
    GsSession.current._put.assert_called_with(
        '/risk/models/{id}/factors/{identifier}'.format(id='id',
                                                        identifier='factor'),
        factor,
        cls=RiskModelFactor)
    assert response == factor
示例#2
0
 def update_factor(self, factor_id: str, factor: RiskModelFactor):
     """ Update existing risk model factor """
     GsRiskModelApi.update_risk_model_factor(self.model.id, factor_id,
                                             factor)