Use NumPy splat instead of list comprehension
Add test setting the matrix diagonal to 1.0