class TestFileReaderClassAndMethods(unittest.TestCase): def setUp(self): self.read_file = FileReader("config.ini") self.read_file1 = FileReader("config_empty.ini") self.read_file2 = FileReader("config_no_rigth_format.ini") self.read_file01 = FileReader("anything.ini") self.read_file02 = FileReader("anything.xls") self.out_put_file_not_exist = '' self.file_empty = " " self.right_format = ['Output_file:Txt\n','Default_alghoritm:Backtracking\n','Dificult_level:Easier\n'] def test_verify_config_file_exist(self): exist_file = self.read_file.file_exist() self.assertEqual(True, exist_file) def test_verify_config_file_is_empty(self): empty_file = self.read_file.file_is_empty() self.assertEqual(False, empty_file) def test_when_config_file_is_empty_return_True(self): empty_file = self.read_file1.file_is_empty() self.assertEqual(True, empty_file) def test_verify_config_file_has_right_extension_ini(self): format_file_name = self.read_file.right_configuration_extension() self.assertTrue(format_file_name) def test_verify_config_file_has_the_new_custom_values(self): custom_setting_file = self.read_file.file_exist() self.assertEqual(True, custom_setting_file) def test_just_right_extension_file_is_read(self): self.assertTrue(self.read_file.right_configuration_extension()) def test_that_a_no_defined_extension_file_is_not_taken_into_account(self): self.assertFalse(self.read_file02.right_configuration_extension()) def test_that_config_file_has_the_right_info_according_to_defined_format(self): read0=self.read_file.right_contain_of_the_ini_file() self.assertEqual(read0, self.right_format) def test_an_error_is_raised_when_there_is_not_file(self): read01=self.read_file01.file_exist() self.failureException(read01) def test_the_right_output_file_value_is_retrived(self): value = self.read_file.read_output_file() self.assertEqual("txt", value.lower()) def test_that_right_difficult_level_is_retrived(self): value = self.read_file.read_dificult_level() self.assertEqual("easier", value.lower()) def test_that_right_default_algrithm_is_retrived(self): value = self.read_file.read_default_algorithm() self.assertEqual("backtracking", value.lower())
class SudokuGenerator: ''' Generator class will create new sudoku games, based on already stored files ''' level = '' files = [] script_dir = os.path.dirname(__file__) path_easy = os.path.join(script_dir, "Sudoku_Levels/Easy/") path_medium = os.path.join(script_dir,'Sudoku_Levels/Medium/') path_hard = os.path.join(script_dir,'Sudoku_Levels/Hard/') def __init__(self,config_file): ''' Constructor: define which is the difficult level set by the user at config.ini file ''' self.readfile = FileReader(config_file) value_level = self.readfile.read_dificult_level() if value_level == 'NoneType': self.level = "" else: self.level = self.readfile.read_dificult_level().lower() def retrieve_file_names(self): """ retrieve_file_names method retrieves all file names stored based on difficult level """ if not self.level: return "No Setting" else: if self.level.strip() == "easier": self.files = [f for f in os.listdir(self.path_easy) if f.endswith('.txt')] return self.files elif self.level.strip() == "medium": self.files = [f for f in os.listdir(self.path_medium) if f.endswith('.txt')] return self.files elif self.level.strip() == "difficult": self.files = [f for f in os.listdir(self.path_hard) if f.endswith('.txt')] return self.files else: return "There is not Sudoku game for this difficult level" def select_file(self): """ Select any file from the relevant difficult set in the configuration file and return a list """ file_list = self.retrieve_file_names() if file_list != "No Setting": num_files = len(file_list) num_f = randint(0, num_files-1) return file_list[num_f] else: return "No Setting" def read_file(self): """ Read the selected file from the list to display """ value = [] files = self.select_file() converter = GeneralConverter() if files == 'No Setting': return files elif self.level.strip() == "easier": value = converter.convert_txt_file_to_string(self.path_easy+'/'+files) elif self.level.strip() == "medium": value = converter.convert_txt_file_to_string(self.path_medium+'/'+files) elif self.level.strip() == "difficult": value = converter.convert_txt_file_to_string(self.path_hard+'/'+files) if len(value) == 81: return converter.convert_string_to_matrix(value)