diff --git a/test/rules_ut.py b/test/rules_ut.py new file mode 100644 index 0000000..29578f6 --- /dev/null +++ b/test/rules_ut.py @@ -0,0 +1,25 @@ +import unittest +from parametrized import parametrized + +from src.rules import Rules + +class TestRules(unittest.TestCase): + @parameterized([ + ('Joe', 7, 6, True), + ('Joe', 10, 10, False), + ('Joe', '7', '6', False), + ('Joe', 7, '1', False), + ('Joe', '1', 6, False), + ]) + def test_check_board_size(self, name, col, row, no_exception=False): + Rules rules + Board board(name, col, row) + + if no_exception: + try: + rules.check_board_sizes(board) + except Exception: + self.assertEqual(no_exception, False) + +if __name__ == "__main__": + unittest.main() \ No newline at end of file