Changed the testing framework to use unittest instead of assert statements