update python unit tests