def test_io_for_diverse_market(self): kospi = stock.get_market_trading_volume_by_date("20200519", "20200520", "KOSPI") kosdaq = stock.get_market_trading_volume_by_date("20200519", "20200520", "KOSDAQ") konex = stock.get_market_trading_volume_by_date("20200519", "20200520", "KONEX") self.assertNotEqual(kospi.iloc[0][1], kosdaq.iloc[0][1]) self.assertNotEqual(kospi.iloc[0][1], konex.iloc[0][1])
def test_io_for_month(self): df = stock.get_market_trading_volume_by_date("20200101", "20200331", "KOSPI", freq="m") self.assertIsNotNone(df) self.assertEqual(len(df), 3)
def test_io_for_kosdaq(self): df = stock.get_market_trading_volume_by_date("20200519", "20200526", "KOSDAQ") print(df) self.assertIsNotNone(df) self.assertIsInstance(df['전체'].iloc[0], np.int64) self.assertIsInstance(df['정규매매'].iloc[0], np.int64)
def test_io_for_kosdaq(self): df = stock.get_market_trading_volume_by_date("20200519", "20200526", "KONEX") self.assertIsNotNone(df) self.assertIsInstance(df['전체'].iloc[0], np.int64) self.assertIsInstance(df['주권'].iloc[0], np.int64) self.assertIsInstance(df['투자회사'].iloc[0], np.int64) self.assertIsInstance(df['부동산투자회사'].iloc[0], np.int64)
def test_trading_value_is_same_3(self): # 기관합계 기타법인 개인 외국인합계 전체 # 날짜 # 2021-01-31 -258249088 1167570 261341862 -4260344 0 df = stock.get_market_trading_volume_by_date("20210115", "20210122", "KOSPI", etf=True, etn=True, elw=True, freq='m') temp = df.iloc[0] == np.array([-258249088, 1167570, 261341862, -4260344, 0]) self.assertEqual(temp.sum(), 5) self.assertIsInstance(df.index , pd.core.indexes.datetimes.DatetimeIndex) self.assertIsInstance(df.index[0], pd._libs.tslibs.timestamps.Timestamp)
def test_trading_value_is_same_2(self): # 기관합계 기타법인 개인 외국인합계 전체 # 날짜 # 2021-01-15 -26571037 8455599 37942108 -19826670 0 # 2021-01-18 -65039501 -757841 39922005 25875337 0 # 2021-01-19 -41855511 4320588 31709703 5825220 0 # 2021-01-20 -23038880 -2562184 38031657 -12430593 0 # 2021-01-21 -38539026 -8798430 38195538 9141918 0 df = stock.get_market_trading_volume_by_date("20210115", "20210122", "KOSPI", etf=True, etn=True, elw=True) temp = df.iloc[0:5, 0] == np.array([-26571037, -65039501, -41855511, -23038880, -38539026]) self.assertEqual(temp.sum(), 5) self.assertIsInstance(df.index , pd.core.indexes.datetimes.DatetimeIndex) self.assertIsInstance(df.index[0], pd._libs.tslibs.timestamps.Timestamp)
def test_trading_value_is_same_1(self): # 기관합계 기타법인 개인 외국인합계 전체 # 날짜 # 2021-01-15 -20393142 8435634 29119751 -17162243 0 # 2021-01-18 -5700054 -1198498 15316328 -8417776 0 # 2021-01-19 7216278 -246496 -24395243 17425461 0 # 2021-01-20 -23038683 -793906 31606917 -7774328 0 # 2021-01-21 -18443990 -7082091 8365421 17160660 0 df = stock.get_market_trading_volume_by_date("20210115", "20210122", "KOSPI") temp = df.iloc[0:5, 0] == np.array([-20393142, -5700054, 7216278, -23038683, -18443990]) self.assertEqual(temp.sum(), 5) self.assertIsInstance(df.index , pd.core.indexes.datetimes.DatetimeIndex) self.assertIsInstance(df.index[0], pd._libs.tslibs.timestamps.Timestamp)
def test_trading_value_is_same_0(self): # 기관합계 기타법인 개인 외국인합계 전체 # 날짜 # 2021-01-15 -5006115 288832 7485785 -2768502 0 # 2021-01-18 505669 262604 151228 -919501 0 # 2021-01-19 1139258 -34023 -2044543 939308 0 # 2021-01-20 -4157919 262408 4917655 -1022144 0 # 2021-01-21 -712099 -321732 2890389 -1856558 0 df = stock.get_market_trading_volume_by_date("20210115", "20210122", "005930") temp = df.iloc[0:5, 0] == np.array([-5006115, 505669, 1139258, -4157919, -712099]) self.assertEqual(temp.sum(), 5) self.assertIsInstance(df.index , pd.core.indexes.datetimes.DatetimeIndex) self.assertIsInstance(df.index[0], pd._libs.tslibs.timestamps.Timestamp) self.assertTrue(df.index[0] < df.index[-1])