Added unit test example in pythn