def test_single_error(self): dw.cleanup_dir() res = dw.download_stats(None) assert res == None save_path = dw.get_saveDir() files = os.listdir(save_path) assert len(files) == 0
def test_all(self): dw.cleanup_dir() dw.download_all() save_path = dw.get_saveDir() files = os.listdir(save_path) assert len(files) == 8 for f in files: assert 'individual' in f or 'team' in f
def test_team(self): dw.cleanup_dir() dw.download_team_stats() save_path = dw.get_saveDir() files = os.listdir(save_path) assert len(files) == 4 for f in files: assert 'team' in f
def test_individual(self): dw.cleanup_dir() dw.download_single_stats() save_path = dw.get_saveDir() files = os.listdir(save_path) assert len(files) == 4 for f in files: assert 'individual' in f
def test_single(self): dw.cleanup_dir() driver = dw.setup_firefox_opt() dw.download_stats(driver) save_path = dw.get_saveDir() files = os.listdir(save_path) assert len(files) == 1 assert 'individual' in files[0] assert 'palka' in files[0]
def read_files(self, category, team_stats = False): dw.cleanup_dir() driver = dw.setup_firefox_opt() dw.download_stats(driver, category, team_stats) driver.close()
def tearDownClass(cls): dw.cleanup_dir()