예제 #1
0
    def test_write_results_failure(self, mock_logger):
        expected_results = OrderedDict()
        expected_results['DIAMOND'] = [(7, 1), (1, 1)]

        ws = WordSearch('data/farm.pzl')
        ws.read()
        ws.solve()
        ws.write()
예제 #2
0
    def test_write_results_valid(self):
        expected_results = [
            'CHICKEN Not Found \n', 'COW (4, 6) (2, 6) \n',
            'PIG (11, 6) (11, 4) \n'
        ]

        ws = WordSearch('data/farm.pzl')
        ws.read()
        ws.solve()
        ws.write()

        with open(ws.file_name + '.out', 'rw') as f:
            data = f.readlines()
            self.assertEquals(data, expected_results)