Simplify python "push" test