Improved confusion matrix test.

Removed hard-coded result matrix for something better.