Adding python unittests