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)