Replaced old unit tests by new ones in python