Switch from crapping test script to actual unit tests