Toeplitz Matrix Python Solution with Tests

lcpy gen -n 766   # by problem number
lcpy gen -s toeplitz_matrix   # by problem name

Input: matrix = [[1,2,3,4],[5,1,2,3],[9,5,1,2]]
Output: true
Explanation:
In the above grid, the diagonals are:
"[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]".
In each diagonal all elements are the same, so the answer is True.

Input: matrix = [[1,2],[2,2]]
Output: false
Explanation:
The diagonal "[1, 2]" has different elements.

from itertools import pairwise


class Solution:
    # Time: O(m * n)
    # Space: O(1)
    def is_toeplitz_matrix(self, matrix: list[list[int]]) -> bool:
        return all(row[:-1] == below[1:] for row, below in pairwise(matrix))