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'))
