def test_index_csv(self): from_date = date(2001,1,15) to_date = date(2001,6,15) raw = nse.index_raw("NIFTY 50", from_date, to_date) output = nse.index_csv("NIFTY 50", from_date, to_date) with open(output) as fp: text = fp.read() rows = [x.split(',') for x in text.split('\n')] assert rows[1][2] == raw[0]['OPEN']
def test_index_df(self): from_date = date(2001,3,15) to_date = date(2001,6,15) raw = nse.index_raw("NIFTY 50", from_date, to_date) df = nse.index_df("NIFTY 50", from_date, to_date) print(df.head()) assert float(raw[0]['OPEN']) == df.iloc[0]['OPEN'] assert float(raw[0]['CLOSE']) == df.iloc[0]['CLOSE'] assert df.iloc[0]['HistoricalDate'] == date(2001,6,15)
def test_index_raw(self): symbol = "NIFTY 50" from_date = date(2020, 6, 1) to_date = date(2020, 7, 30) d = nse.index_raw(symbol, from_date, to_date) assert d[0]['Index Name'] == 'Nifty 50' assert d[0]['HistoricalDate'] == '30 Jul 2020' assert d[-1]['HistoricalDate'] == '01 Jun 2020' app_name = nse.APP_NAME + '-index' files = os.listdir(user_cache_dir(app_name, app_name)) assert len(files) == 2