Python GsDataApi.get_many_coordinatesの例

コード例 #1

 def coordinates_for_asset(result: DataFrameWithInfo):
     return tuple(
         itertools.chain.from_iterable(
             GsDataApi.get_many_coordinates(
                 mkt_type=t,
                 mkt_asset=a,
                 return_type=MarketDataCoordinate,
                 limit=10000)
             for t, a in zip(result.mkt_type, result.mkt_asset)))

コード例 #2

ファイル: test_data.py プロジェクト: yanglp904921/gs-quant

def test_get_many_coordinates(mocker):
    coordinates = [{
        'id': 'MC123',
        'name': 'A_B_C_D_E.F1'
    }, {
        'id': 'MC123',
        'name': 'A_B_C_D_E.F2'
    }]
    mocker.patch.object(GsSession.__class__,
                        'default_value',
                        return_value=GsSession.get(Environment.QA, 'client_id',
                                                   'secret'))
    GsSession.current._post = mocker.Mock(
        return_value={'results': coordinates})
    response = GsDataApi.get_many_coordinates(mkt_type='A', mkt_asset='B')
    assert response == ('A_B_C_D_E.F1', 'A_B_C_D_E.F2')