[WIP] Improve python test coverage