Make tests for non-diagonal covariance matrix

[DeaglanBartlett]

Currently the unit tests only consider a diagonal covariance matrix, even when using the matrix form. This should be made more general