Add unit test for diagonal_indices()