def test_trading_volume_is_same_1(self):
# 매도 매수 순매수
# 투자자구분
# 금융투자 137969209 127697577 -10271632
# 보험 15737709 8577242 -7160467
# 투신 46872846 34307243 -12565603
# 사모 20780475 16342937 -4437538
# 은행 2236667 632814 -1603853
df = stock.get_market_trading_volume_by_investor("20210115", "20210122", "KOSPI")
temp = df.iloc[0:5, -1] == np.array([-10271632, -7160467, -12565603, -4437538, -1603853])
self.assertEqual(temp.sum(), 5)
def test_trading_volume_is_same_2(self):
# 매도 매수 순매수
# 투자자구분
# 금융투자 1857447354 1660620713 -196826641
# 보험 29594468 19872165 -9722303
# 투신 69348190 60601427 -8746763
# 사모 31673292 26585281 -5088011
# 은행 44279242 51690814 7411572
df = stock.get_market_trading_volume_by_investor("20210115", "20210122", "KOSPI", etf=True, etn=True, elw=True)
temp = df.iloc[0:5, -1] == np.array([-196826641, -9722303, -8746763, -5088011, 7411572])
self.assertEqual(temp.sum(), 5)
def test_trading_volume_is_same_0(self):
# 매도 매수 순매수
# 투자자구분
# 금융투자 29455909 26450600 -3005309
# 보험 1757287 509535 -1247752
# 투신 2950680 1721970 -1228710
# 사모 745727 696135 -49592
# 은행 38675 46394 7719
df = stock.get_market_trading_volume_by_investor("20210115", "20210122", "005930")
temp = df.iloc[0:5, -1] == np.array([-3005309, -1247752, -1228710, -49592, 7719])
self.assertEqual(temp.sum(), 5)
temp = df.index[0:5] == np.array(["금융투자", "보험", "투신", "사모", "은행"])
self.assertEqual(temp.sum(), 5)