Refactor tests to use unittest

Reviewers: marcua

Reviewed By: marcua

Differential Revision: http://phabricator.locu.com/D3311