I added a single simple test, but I'm actually unsure of how to go about testing for correctness...
- Is there perhaps a way to generate a ground truth data set for a non-trivial example?