def test_gdshe_data_no_properties(self, mocked_post): ciq_client = CapIQClient("username", "password") return_value = ciq_client.gdshe(["TRIP"], ["IQ_CLOSEPRICE"], ["close_price"], start_date="12/19/1980", end_date="12/19/2000") self.assertEqual(return_value, {'TRIP:': {'close_price': [['46.80']]}})
def test_gdshe_multi_data_with_name_collision(self, mocked_post): ciq_client = CapIQClient("username", "password") return_value = ciq_client.gdshe(["TRIP", "TRIP"], ["IQ_VWAP", "IQ_VWAP"], ["vwap1", "vwap2"], properties=[{ "STARTDATE": "05/23/2017", "ENDDATE": "05/29/2017" }, { "STARTDATE": "05/16/2017", "ENDDATE": "05/22/2017" }]) self.assertEqual( return_value, { 'TRIP:': { 'vwap1': [['239.900000', '5/23/2017'], ['240.410000', '5/24/2017'], ['241.710000', '5/25/2017'], ['241.590000', '5/26/2017']], 'vwap2': [['240.040000', '5/16/2017'], ['237.870000', '5/17/2017'], ['236.640000', '5/18/2017'], ['238.330000', '5/19/2017'], ['239.210000', '5/22/2017']] } })
def test_gdshe_no_data(self, mocked_post): ciq_client = CapIQClient("username", "password") return_value = ciq_client.gdshe(["TRIP"], ["IQ_CLOSEPRICE"], ["close_price"], [{}]) self.assertEqual(return_value, {'TRIP:': {'close_price': None}})