def setUp(self):
self.test_data = {
'well_formed_content' : "067040002\n" +
"900730450\n" +
"405000000\n" +
"600080000\n" +
"000900320\n" +
"283500649\n" +
"070000003\n" +
"000657804\n" +
"006423597\n",
'file_with_spaces' : "067 040 002\n" +
"900 730 450\n" +
"405 000 000\n" +
"600 080 000\n" +
"000 900 320\n" +
"283 500 649\n" +
"070 000 003\n" +
"000 657 804\n" +
"006 423 597\n",
'file_with_chars' : "67.4...2\n" +
"9..73.45.\n" +
"4.5......\n" +
"6...8....\n" +
"...9..32.\n" +
"2835..649\n" +
".7......3\n" +
"...6578.4\n" +
"..6423597\n",
'file_contains_more_than_9_lines' : "067040002\n" +
"900730450\n" +
"405000000\n" +
"600080000\n" +
"000900320\n" +
"283500649\n" +
"070000003\n" +
"000657804\n" +
"006423597\n" +
"\n",
'file_contains_more_than_9_cols' : "067040002\n" +
"900730450\n" +
"4050000000\n" +
"600080000\n" +
"000900320\n" +
"283500649\n" +
"070000003\n" +
"000657804\n" +
"006423597\n"
}
self.extension = ".txt"
self.test_file_names = {}
self.txtparser = TXTParser()
self.puzzle = {
"A1":'.', "A2":'6', "A3":'7', "A4":'.', "A5":'4', "A6":'.', "A7":'.', "A8":'.', "A9":'2',
"B1":'9', "B2":'.', "B3":'.', "B4":'7', "B5":'3', "B6":'.', "B7":'4', "B8":'5', "B9":'.',
"C1":'4', "C2":'.', "C3":'5', "C4":'.', "C5":'.', "C6":'.', "C7":'.', "C8":'.', "C9":'.',
"D1":'6', "D2":'.', "D3":'.', "D4":'.', "D5":'8', "D6":'.', "D7":'.', "D8":'.', "D9":'.',
"E1":'.', "E2":'.', "E3":'.', "E4":'9', "E5":'.', "E6":'.', "E7":'3', "E8":'2', "E9":'.',
"F1":'2', "F2":'8', "F3":'3', "F4":'5', "F5":'.', "F6":'.', "F7":'6', "F8":'4', "F9":'9',
"G1":'.', "G2":'7', "G3":'.', "G4":'.', "G5":'.', "G6":'.', "G7":'.', "G8":'.', "G9":'3',
"H1":'.', "H2":'.', "H3":'.', "H4":'6', "H5":'5', "H6":'7', "H7":'8', "H8":'.', "H9":'4',
"I1":'.', "I2":'.', "I3":'6', "I4":'4', "I5":'2', "I6":'3', "I7":'5', "I8":'9', "I9":'7'
}
for file_name, content in self.test_data.items():
with open(file_name + self.extension, 'w') as test_file:
test_file.write(content)
self.test_file_names[file_name] = file_name + self.extension
with open('not_a_txt_file.dat', 'w') as not_a_txt_file:
not_a_txt_file.write('Dummy text.dat')
self.test_file_names['not_a_txt_file'] = 'not_a_txt_file.dat'
class TestTXTParser(unittest.TestCase):
def setUp(self):
self.test_data = {
'well_formed_content' : "067040002\n" +
"900730450\n" +
"405000000\n" +
"600080000\n" +
"000900320\n" +
"283500649\n" +
"070000003\n" +
"000657804\n" +
"006423597\n",
'file_with_spaces' : "067 040 002\n" +
"900 730 450\n" +
"405 000 000\n" +
"600 080 000\n" +
"000 900 320\n" +
"283 500 649\n" +
"070 000 003\n" +
"000 657 804\n" +
"006 423 597\n",
'file_with_chars' : "67.4...2\n" +
"9..73.45.\n" +
"4.5......\n" +
"6...8....\n" +
"...9..32.\n" +
"2835..649\n" +
".7......3\n" +
"...6578.4\n" +
"..6423597\n",
'file_contains_more_than_9_lines' : "067040002\n" +
"900730450\n" +
"405000000\n" +
"600080000\n" +
"000900320\n" +
"283500649\n" +
"070000003\n" +
"000657804\n" +
"006423597\n" +
"\n",
'file_contains_more_than_9_cols' : "067040002\n" +
"900730450\n" +
"4050000000\n" +
"600080000\n" +
"000900320\n" +
"283500649\n" +
"070000003\n" +
"000657804\n" +
"006423597\n"
}
self.extension = ".txt"
self.test_file_names = {}
self.txtparser = TXTParser()
self.puzzle = {
"A1":'.', "A2":'6', "A3":'7', "A4":'.', "A5":'4', "A6":'.', "A7":'.', "A8":'.', "A9":'2',
"B1":'9', "B2":'.', "B3":'.', "B4":'7', "B5":'3', "B6":'.', "B7":'4', "B8":'5', "B9":'.',
"C1":'4', "C2":'.', "C3":'5', "C4":'.', "C5":'.', "C6":'.', "C7":'.', "C8":'.', "C9":'.',
"D1":'6', "D2":'.', "D3":'.', "D4":'.', "D5":'8', "D6":'.', "D7":'.', "D8":'.', "D9":'.',
"E1":'.', "E2":'.', "E3":'.', "E4":'9', "E5":'.', "E6":'.', "E7":'3', "E8":'2', "E9":'.',
"F1":'2', "F2":'8', "F3":'3', "F4":'5', "F5":'.', "F6":'.', "F7":'6', "F8":'4', "F9":'9',
"G1":'.', "G2":'7', "G3":'.', "G4":'.', "G5":'.', "G6":'.', "G7":'.', "G8":'.', "G9":'3',
"H1":'.', "H2":'.', "H3":'.', "H4":'6', "H5":'5', "H6":'7', "H7":'8', "H8":'.', "H9":'4',
"I1":'.', "I2":'.', "I3":'6', "I4":'4', "I5":'2', "I6":'3', "I7":'5', "I8":'9', "I9":'7'
}
for file_name, content in self.test_data.items():
with open(file_name + self.extension, 'w') as test_file:
test_file.write(content)
self.test_file_names[file_name] = file_name + self.extension
with open('not_a_txt_file.dat', 'w') as not_a_txt_file:
not_a_txt_file.write('Dummy text.dat')
self.test_file_names['not_a_txt_file'] = 'not_a_txt_file.dat'
def tearDown(self):
for file_name in self.test_file_names.values():
try:
os.remove(file_name)
except IOError:
continue
def test_a_well_formed_txt_file_should_be_parsed(self):
actual_results = self.txtparser.parse_puzzle(self.test_file_names['well_formed_content'])
expected_results = self.puzzle
self.assertEqual(expected_results, actual_results)
def test_a_SyntaxError_should_be_raised_if_file_contains_spaces(self):
with self.assertRaises(SyntaxError):
self.txtparser.parse_puzzle(self.test_file_names['file_with_spaces'])
def test_a_SyntaxError_should_be_raised_if_file_contains_non_digit_characters(self):
with self.assertRaises(SyntaxError):
self.txtparser.parse_puzzle(self.test_file_names['file_with_chars'])
def test_a_SyntaxError_should_be_raised_if_file_contains_more_than_nine_lines(self):
with self.assertRaises(SyntaxError):
self.txtparser.parse_puzzle(self.test_file_names['file_contains_more_than_9_lines'])
def test_a_SyntaxError_should_be_raised_if_file_contains_more_than_nine_columns(self):
with self.assertRaises(SyntaxError):
self.txtparser.parse_puzzle(self.test_file_names['file_contains_more_than_9_cols'])
def test_an_FileFormatError_should_be_raised_if_file_extension_is_not_txt(self):
with self.assertRaises(FileFormatError):
self.txtparser.parse_puzzle(self.test_file_names['not_a_txt_file'])
def test_an_IOError_should_be_raised_if_file_does_not_exist(self):
with self.assertRaises(IOError):
self.txtparser.parse_puzzle('missing_file.txt')
@unittest.skip("Need to find a better way to test reading a puzzle file when access is denied")
def test_an_IOError_should_be_raised_if_access_to_file_is_denied(self):
with self.assertRaises(IOError):
self.txtparser.parse_puzzle('C:\\TestFolder\\puzzle.txt')