Added python unittests for models