def test_init_root(self):
dirname = 'testinit'
assert getcwd() == self.cwd
FileIO.init_root(dirname, False)
assert getcwd() == urljoin(self.cwd+'/', dirname)
chdir('..')
assert path.exists(dirname) is True
rmdir(dirname)
def __scrap_categories(self, to_csv=False):
FileIO.init_root('data', False)
categories = []
progress_monitor.allbooks_init(self.num_books, self.site_url)
for link in self.links:
progress_monitor.category_update(
len(categories),
len(self.links),
link[1])
category = Category(link[0])
categories.append(category)
FileIO.open_category(category.name)
category.write_csv()
FileIO.close_category()
return categories