def test_not_empty_result(self): df = stock.get_shorting_investor_volume_by_date( "20190401", "20190405", "KOSPI") self.assertNotEqual(df.empty, True) df = stock.get_shorting_investor_price_by_date("20190401", "20190405", "KOSPI") self.assertNotEqual(df.empty, True) df = stock.get_shorting_volume_top50("20190401", "KOSPI") self.assertNotEqual(df.empty, True) df = stock.get_shorting_balance_top50("20190401", "KOSPI") self.assertNotEqual(df.empty, True)
def test_in_kosdaq_1(self): df = stock.get_shorting_volume_top50("20200106", "KOSDAQ") self.assertIsInstance(df, pd.DataFrame) self.assertTrue(len(df), 50)
def test_with_holiday(self): # 20200105 sunday df = stock.get_shorting_volume_top50("20200105") self.assertIsInstance(df, pd.DataFrame) self.assertTrue(df.empty)
def test_with_default_param(self): df = stock.get_shorting_volume_top50("20200106") self.assertIsInstance(df, pd.DataFrame) self.assertTrue(len(df), 50)