Modify test_fill_diagonal_offset.py to cover every possible case