def test_trading_value_is_same_1(self): # 매도 매수 순매수 # 투자자구분 # 금융투자 9827334289654 9294592831462 -532741458192 # 보험 912820396542 560818697065 -352001699477 # 투신 1790231574897 1421181450288 -369050124609 # 사모 830445404788 665802837480 -164642567308 # 은행 58624439870 37109603010 -21514836860 df = stock.get_market_trading_value_by_investor("20210115", "20210122", "KOSPI") temp = df.iloc[0:5, -1] == np.array([-532741458192, -352001699477, -369050124609, -164642567308, -21514836860]) self.assertEqual(temp.sum(), 5)
def test_trading_value_is_same_2(self): # 매도 매수 순매수 # 투자자구분 # 금융투자 15985568261831 15006116511544 -979451750287 # 보험 1219035502445 757575677208 -461459825237 # 투신 2235561259511 1799363743367 -436197516144 # 사모 999084910863 846067212945 -153017697918 # 은행 886226324790 936210985810 49984661020 df = stock.get_market_trading_value_by_investor("20210115", "20210122", "KOSPI", etf=True, etn=True, elw=True) temp = df.iloc[0:5, -1] == np.array([-979451750287, -461459825237, -436197516144, -153017697918, 49984661020]) self.assertEqual(temp.sum(), 5)
def test_trading_value_is_same_0(self): # 매도 매수 순매수 # 투자자구분 # 금융투자 2580964135000 2309054317700 -271909817300 # 보험 153322228800 44505136200 -108817092600 # 투신 258073006600 150715203700 -107357802900 # 사모 65167773900 60862926800 -4304847100 # 은행 3369626100 4004806100 635180000 df = stock.get_market_trading_value_by_investor("20210115", "20210122", "005930") temp = df.iloc[0:5, -1] == np.array([-271909817300, -108817092600, -107357802900, -4304847100, 635180000]) self.assertEqual(temp.sum(), 5) temp = df.index[0:5] == np.array(["금융투자", "보험", "투신", "사모", "은행"]) self.assertEqual(temp.sum(), 5)