Update tests to use unittest2 methods instead of bare asserts