def test_retrieve_data(self): df = YahooStock.daily_k_line('sh600010', method=self.get_mock_input) assert_that(df.columns.name, equal_to('sh600010')) assert_that(len(df), equal_to(5)) record = df.ix[20150206] assert_that(record.ix['open'], equal_to(4.94)) assert_that(record.ix['adjClose'], equal_to(4.74))
def test_index_converter_sz(self): yahoo_index = YahooStock._index_converter('sz000010') assert_that(yahoo_index, equal_to('000010.sz'))
def test_index_converter_sh(self): yahoo_index = YahooStock._index_converter('sh600010') assert_that(yahoo_index, equal_to('600010.ss'))