rewrite tests for multiple python versions