Improve python test coverage