class OneRecordTest(unittest.TestCase): def setUp(self): #self.browser = webdriver.Chrome() self.crawler = Crawler() pass def setDown(self): #self.browser.quit() pass def test_find_records(self): # 開啟瀏覽器 self.crawler.open_browser() # 讀取第一筆清單上的名稱 list.csv records = self.crawler.get_records("./list.csv") self.assertEqual(len(records),2)