blocks.py

import numpy as np

BLOCKS = [
    # 1 tile
    np.array([[1]]),
    # 2 tiles
    np.array([[1, 1]]),
    # 3 tiles
    np.array([[1, 1], [0, 1]]),
    np.array([[1, 1, 1]]),
    # 4 tiles
    np.array([[1, 1, 1, 1]]),
    np.array([[1, 1, 1], [0, 0, 1]]),
    np.array([[1, 1, 0], [0, 1, 1]]),
    np.array([[1, 1], [1, 1]]),
    np.array([[1, 1, 1], [0, 1, 0]]),
    # 5 tiles
    np.array([[0, 1, 1], [1, 1, 0], [0, 1, 0]]),
    np.array([[1, 1, 1, 1, 1]]),
    np.array([[1, 1, 1, 1], [1, 0, 0, 0]]),
    np.array([[0, 1, 1, 1], [1, 1, 0, 0]]),
    np.array([[1, 1, 1], [1, 1, 0]]),
    np.array([[1, 1, 1], [0, 1, 0], [0, 1, 0]]),
    np.array([[1, 1, 1], [1, 0, 1]]),
    np.array([[1, 1, 1], [1, 0, 0], [1, 0, 0]]),
    np.array([[0, 0, 1], [0, 1, 1], [1, 1, 0]]),
    np.array([[0, 1, 0], [1, 1, 1], [0, 1, 0]]),
    np.array([[1, 1, 1, 1], [0, 1, 0, 0]]),
    np.array([[1, 1, 0], [0, 1, 0], [0, 1, 1]]),
]