from Python import do_continue
from PythonPersonalProject.Python.models.chessboard import ChessBoard, check_valid_number

if __name__ == '__main__':
    is_continue = True
    while is_continue:
        print(
            'Enter height and width of chessboard. Values must be only integer numbers.'
        )
        try:
            entered_value = input('Height: ')
            height = check_valid_number(entered_value)
            entered_value = input('Width: ')
            width = check_valid_number(entered_value)
        except ValueError:
            print('Incorrect input.')
            continue
        chessboard = ChessBoard(height, width)
        chessboard.print_chessboard()
        response = str(input('Do you want to continue? [y / yes]\n'))
        is_continue = do_continue(response)
    print('The program is completed.')
 def test_check_valid_number_1(self):
     self.assertEqual(chessboard.check_valid_number(67), 67)
 def test_check_valid_number_6(self):
     self.assertEqual(chessboard.check_valid_number(''), 0)
 def test_check_valid_number_0(self):
     self.assertEqual(chessboard.check_valid_number(8), 8)
 def test_check_valid_number_3(self):
     self.assertEqual(chessboard.check_valid_number(1), 1)