[IMP]Improved python test cases.