def test_with_4_strig_and_sorting(self): self.create_file_for_check( "ccc_bbb_aaa\naaa_ccc_bbb\nbbb_ccc_aaa\naaa_bbb_ccc") multisorter = Sorter(input_file_name="test.txt", separators="_", strings_in_tmp_file=2, is_bar_need=False) stable_sorter = Sorter(input_file_name="test.txt", separators="_", is_multisorting=False, column_for_sort=2, strings_in_tmp_file=2, is_bar_need=False) multisorter.split_file_into_sorted_tmp_files() os.chdir(multisorter.path_tmp_dir) tmp_file_0 = open("0.tmp", "r") tmp_file_1 = open("1.tmp", "r") self.assertEqual("aaa_ccc_bbb\nccc_bbb_aaa", tmp_file_0.read()) self.assertEqual("aaa_bbb_ccc\nbbb_ccc_aaa", tmp_file_1.read()) tmp_file_0.close() tmp_file_1.close() os.chdir(stable_sorter.path_tmp_dir) stable_sorter.split_file_into_sorted_tmp_files() tmp_file_0 = open("0.tmp", "r") tmp_file_1 = open("1.tmp", "r") self.assertEqual("bbb_0_ccc_bbb_aaa\nccc_1_aaa_ccc_bbb", tmp_file_0.read()) self.assertEqual("bbb_3_aaa_bbb_ccc\nccc_2_bbb_ccc_aaa", tmp_file_1.read()) tmp_file_0.close() tmp_file_1.close() self.assertEqual( len(stable_sorter.tmp_files_names), 2 )
def test_empty_file(self): self.create_file_for_check("") sorter = Sorter(input_file_name="test.txt", separators="_", is_bar_need=False) sorter.split_file_into_sorted_tmp_files() os.chdir(sorter.path_tmp_dir) tmp_file_0 = open("0.tmp", "r") self.assertEqual("", tmp_file_0.read()) tmp_file_0.close()
def test_file_with_single_string(self): self.create_file_for_check("1_2_345_a") multisorter = Sorter(input_file_name="test.txt", separators="_", is_bar_need=False) stable_sorter = Sorter(input_file_name="test.txt", separators="_", is_multisorting=False, column_for_sort=2, is_bar_need=False) multisorter.split_file_into_sorted_tmp_files() os.chdir(multisorter.path_tmp_dir) tmp_file_0 = open("0.tmp", "r") self.assertEqual("1_2_345_a", tmp_file_0.read()) tmp_file_0.close() stable_sorter.split_file_into_sorted_tmp_files() os.chdir(stable_sorter.path_tmp_dir) tmp_file_0 = open("0.tmp", "r") self.assertEqual("2_0_1_2_345_a", tmp_file_0.read()) tmp_file_0.close()