Unit test strictly triangular matrix Python