CDAP-488 Improve integration tests for python

Ignore this comment. Its OK as is.