diff --git a/tests/fetchers/test_akshare.py b/tests/fetchers/test_akshare.py index 2c489e2..33f4492 100644 --- a/tests/fetchers/test_akshare.py +++ b/tests/fetchers/test_akshare.py @@ -7,8 +7,8 @@ def test_fetch_stock_list(self): df = fetch_stock_list() self.assertTrue(len(df) >= 1000) self.assertSetEqual(set(df["type"]), set(["stock", "index"])) - filter = df[df.code == "000003.SZ"] + filter = df.code == "000003.SZ" self.assertEqual(df[filter].iloc[0]["exit"], datetime.date(2002, 6, 14)) - filter = df[df.code == "000001.SH"] + filter = df.code == "000001.SH" self.assertEqual(df[filter].iloc[0]["alias"], "上证指数")