def test_in_kosdaq_1(self):
df = stock.get_shorting_volume_by_ticker("20210104", "KOSDAQ")
# 공매도 매수 비중
# 티커
# 060310 0 588133 0.000000
# 054620 0 300316 0.000000
# 265520 0 415285 0.000000
# 211270 0 424730 0.000000
# 035760 1042 156069 0.669922
self.assertIsInstance(df, pd.DataFrame)
temp = df.index[:5] == np.array(
["060310", "054620", "265520", "211270", "035760"])
self.assertEqual(temp.sum(), 5)
def test_in_konex(self):
df = stock.get_shorting_volume_by_ticker("20210104", market="KONEX")
# 공매도 매수 비중
# 티커
# 112190 0 1240 0.0
# 224880 0 0 0.0
# 183410 0 0 0.0
# 076340 0 7933 0.0
# 329050 0 0 0.0
self.assertIsInstance(df, pd.DataFrame)
temp = df.index[:5] == np.array(
["112190", "224880", "183410", "076340", "329050"])
self.assertEqual(temp.sum(), 5)
def test_with_default_param(self):
df = stock.get_shorting_volume_by_ticker("20210125")
# 공매도 매수 비중
# 티커
# 095570 32 180458 0.020004
# 006840 79 386257 0.020004
# 027410 18502 8453962 0.219971
# 282330 96 82986 0.119995
# 138930 1889 1181748 0.160034
self.assertIsInstance(df, pd.DataFrame)
temp = df.iloc[0:5, 0] == np.array([32, 79, 18502, 96, 1889])
self.assertEqual(temp.sum(), 5)
temp = df.index[0:5] == np.array(
["095570", "006840", "027410", "282330", "138930"])
self.assertEqual(temp.sum(), 5)
def test_with_holiday(self):
df_0 = stock.get_shorting_volume_by_ticker("20210103")
df_1 = stock.get_shorting_volume_by_ticker("20201230")
same = (df_0 == df_1).all(axis=None)
self.assertTrue(same)
