adding python unit tests