diff --git a/field.py b/field.py index bda068b..f212fdc 100644 --- a/field.py +++ b/field.py @@ -69,7 +69,7 @@ class Field(): assert column >= 0 assert column + tetromino.width() <= Field.WIDTH assert row - tetromino.height() + 1 >= 0 - assert row < (Field.HEIGHT / 2) + assert row < Field.HEIGHT for ti, si in list(enumerate(range(row - tetromino.height() + 1, row + 1)))[::-1]: for tj, sj in enumerate(range(column, column + tetromino.width())): @@ -87,7 +87,7 @@ class Field(): assert column >= 0 assert column + tetromino.width() <= Field.WIDTH assert row - tetromino.height() + 1 >= 0 - assert row < (Field.HEIGHT / 2) + assert row < Field.HEIGHT for ti, si in list(enumerate(range(row - tetromino.height() + 1, row + 1)))[::-1]: for tj, sj in enumerate(range(column, column + tetromino.width())): @@ -105,7 +105,9 @@ class Field(): assert column >= 0 assert column + tetromino.width() <= Field.WIDTH last_fit = -1 - for row in range(tetromino.height(), (Field.HEIGHT / 2)): + for row in range(tetromino.height(), Field.HEIGHT): + if Field.HEIGHT <= Field.HEIGHT/2: + continue if self._test_tetromino(tetromino, row, column): last_fit = row else: