async def evaluate_handler( item: EvaluateRequest, request: Request = Depends(AuthorizationRequired()), Authorization: str = Header(..., description="Bearer your_key"), ): result = evaluate.evaluate_handler(Authorization, item.language) if result.get("status") and result.get("error"): raise HTTPException(status_code=400, detail=result) return result
async def evaluate_handler( language: str = Form(...), repository_version: Optional[int] = Form(default=None), request: Request = Depends(AuthorizationRequired()), Authorization: str = Header(..., description="Bearer your_key"), ): result = evaluate.evaluate_handler(Authorization, language, repository_version) if result.get("status") and result.get("error"): raise HTTPException(status_code=400, detail=result) return result
def test_invalid_auth(self): with self.assertRaises(AuthorizationIsRequired): evaluate_handler("", "en", self.repository_version) with self.assertRaises(AuthorizationIsRequired): evaluate_handler(None, "en", self.repository_version) with self.assertRaises(AuthorizationIsRequired): evaluate_handler(2, "en", self.repository_version) with self.assertRaises(AuthorizationIsRequired): crossvalidation_evaluate_handler("", "en", self.repository_version) with self.assertRaises(AuthorizationIsRequired): crossvalidation_evaluate_handler(None, "en", self.repository_version) with self.assertRaises(AuthorizationIsRequired): crossvalidation_evaluate_handler(2, "en", self.repository_version)
def test_invalid_lang(self): with self.assertRaises(ValidationError): evaluate_handler( self.authorization, "unknown_lang", self.repository_version ) with self.assertRaises(ValidationError): evaluate_handler(self.authorization, None, self.repository_version) with self.assertRaises(ValidationError): evaluate_handler(self.authorization, 2, self.repository_version) with self.assertRaises(ValidationError): crossvalidation_evaluate_handler( self.authorization, "unknown_lang", self.repository_version ) with self.assertRaises(ValidationError): crossvalidation_evaluate_handler( self.authorization, None, self.repository_version ) with self.assertRaises(ValidationError): crossvalidation_evaluate_handler( self.authorization, 2, self.repository_version )
def test_evaluate_untrained_model(self, *args): with self.assertRaises(ValidationError): evaluate_handler(self.authorization, "en", self.repository_version)
def test_evaluate_mocked_celery(self, *args): evaluate_handler(self.authorization, "en", self.repository_version)
def test_evaluate_mocked_celery_timeout(self, *args): with self.assertRaises(CeleryTimeoutException): evaluate_handler(self.authorization, "en", self.repository_version)