def df(): user_id = os.environ.get("FINMIND_USER", "") password = os.environ.get("FINMIND_PASSWORD", "") data_loader = DataLoader() data_loader.login(user_id, password) df = data_loader.taiwan_stock_holding_shares_per(stock_id="2890", start_date="2021-06-01", end_date="2021-07-03") df["labels"] = df["HoldingSharesLevel"] df["series"] = df["percent"] return df
def test_pie(stock_id, start_date, end_date): user_id = os.environ.get("FINMIND_USER", "") password = os.environ.get("FINMIND_PASSWORD", "") data_loader = DataLoader() data_loader.login(user_id, password) df = data_loader.taiwan_stock_holding_shares_per(stock_id=stock_id, start_date=start_date, end_date=end_date) df = df[df["date"] == max(df["date"])] df = df[df["HoldingSharesLevel"] != "total"] df["labels"] = df["HoldingSharesLevel"] df["series"] = df["percent"] assert plotting.pie(labels=df["labels"], series=df["series"])