It's good practice to use a name that's a bit more descriptive. Note that the other tests use `i` and `j` but those are arrays of sequential row/column indices.

Also, PEP8: spaces after commas.