Better Testing?

Currently, the testing just makes sure that the algorithm runs without error on a toy problem. 

It would be nice to do something more akin to unit testing, but I'm not quite sure how to do it in this situation since a lot of the testable functionality is provided by the user.