def test_split_by_month(self):
tmp_dir = os.path.join(get_test_dir(), "tmp_split_month")
if not os.path.exists(tmp_dir):
os.makedirs(tmp_dir)
split_by_month(csv_file, tmp_dir, verbose=False)
self._test_bm(csv_file, tmp_dir)
shutil.rmtree(tmp_dir)
def test_parallel_fix_date(self):
tmp_dir = os.path.join(get_test_dir(), "tmp_fix_date")
if not os.path.exists(tmp_dir):
os.makedirs(tmp_dir)
out_file = os.path.join(tmp_dir, "out.csv")
parallel_fix_date(csv_file, out_file)
self._test_fd(csv_file, out_file)
shutil.rmtree(tmp_dir)
def test_parallel_split_by_month(self):
tmp_dir = os.path.join(get_test_dir(), "tmp_parallel_split_month")
parallel_split_by_month(csv_file, tmp_dir, nworkers=10)
self._test_bm(csv_file, tmp_dir)
shutil.rmtree(tmp_dir)