added python unit tests