Stop using unittest assert statements from 2.7+