tests: Lineage tracing with python

[Janardhan]

Hi Kevin,

The lineage tests were failing as mentioned in this comment.
https://github.com/apache/systemml/pull/873#issuecomment-617799527

Shall we fix these with priority. Thanks.

Thanks,
Janardhan