CDAP-488 Improve integration tests for python

Same here. And the line below.