CDAP-488 Improve integration tests for python

@anwar6953 hm, on my machine vise versa, old statement is not working