Fix test assertions for python 2.6