Extend the test dataset to support all regression types

[daniel-thom]

Should we expand the test dataset so this can be uncommented?

Originally posted by @elainethale in #7 (comment)