Python Dataset.equalsの例

コード例 #1

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

def test_get_coverage(mocker):
    mocker.patch("gs_quant.api.gs.data.GsDataApi.get_coverage",
                 return_value=test_coverage_data)
    mocker.patch("gs_quant.api.gs.data.GsDataApi.get_types",
                 return_value={'gsid': 'string'})
    data = Dataset(Dataset.TR.TREOD).get_coverage()
    assert data.equals(
        GsDataApi.construct_dataframe_with_types(str(Dataset.TR.TREOD),
                                                 test_coverage_data))

コード例 #2

def test_get_coverage(get_coverage, get_types):
    get_coverage.return_value = test_coverage_data
    get_types.return_value = {'gsid': 'string'}
    data = Dataset(Dataset.TR.TREOD).get_coverage()

    assert data.equals(construct_dataframe_with_types(str(Dataset.TR.TREOD), test_coverage_data))