Skip to content

Commit 4fa45a0

Browse files
committed
Made test files location more generic.
1 parent a7e4553 commit 4fa45a0

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

tests/test_sudoku.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,21 +19,27 @@
1919
from __future__ import unicode_literals
2020
from __future__ import absolute_import
2121

22+
import os
23+
2224
from hbldhdoku.sudoku import Sudoku
2325
from hbldhdoku.exceptions import SudokuException
2426

2527

2628
class TestSudoku(object):
27-
"""Test Suite for Sukoku solver."""
29+
"""Test Suite for Sudoku solver."""
30+
31+
def __init__(self):
32+
self.test_dir = os.path.dirname(os.path.abspath(__file__))
2833

2934
def test_solve_simple_sudoku(self):
30-
s = Sudoku.load_sudoku('./simple.sud')
35+
36+
s = Sudoku.load_sudoku(os.path.join(self.test_dir, 'simple.sud'))
3137
s.solve()
32-
correct_solution = Sudoku.load_sudoku('./simple_sol.sud')
38+
correct_solution = Sudoku.load_sudoku(os.path.join(self.test_dir, 'simple_sol.sud'))
3339
assert s == correct_solution
3440

3541
def test_solve_medium_sudoku(self):
36-
s = Sudoku.load_sudoku('./medium.sud')
42+
s = Sudoku.load_sudoku(os.path.join(self.test_dir, 'medium.sud'))
3743
s.solve()
38-
correct_solution = Sudoku.load_sudoku('./medium_sol.sud')
44+
correct_solution = Sudoku.load_sudoku(os.path.join(self.test_dir, 'medium_sol.sud'))
3945
assert s == correct_solution

0 commit comments

Comments
 (0)