コード例 #1
0
ファイル: function_test.py プロジェクト: ntuaha/Medicine
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)