MAINT: setdiag edge cases and enable coo setdiag tests