Improve Tests

[akashsaravanan-georgian]

Currently the tests just check if every model can run on a set of predetermined dataset configurations. Add additional tests to check scenarios where we have no text, no categorical or no numerical columns so that issues like #82 are avoided.