CDAP-488 Improve integration tests for python

Does this work on your machine, or am I missing something?