added test cases for Diagonal