Update test_matrices.py