test with small iteration and smaller matrix shape. Also, test with matrix not square whenever possible.