Add test for blkdiag