Method create_prediction_result

tests/conftest.py:180–205

Create a PredictionResult for testing.

(
        text: str, entities: List[Dict], model_name: str = "test-model"
    )

    @staticmethod
    def create_prediction_result(
        text: str, entities: List[Dict], model_name: str = "test-model"
    ):
        """Create a PredictionResult for testing."""
        from datetime import datetime

        from openmed.processing.outputs import EntityPrediction, PredictionResult

        entity_objects = [
            EntityPrediction(
                text=e["word"],
                label=e["entity"],
                confidence=e["score"],
                start=e.get("start"),
                end=e.get("end"),
            )
            for e in entities
        ]

        return PredictionResult(
            text=text,
            entities=entity_objects,
            model_name=model_name,
            timestamp=datetime.now().isoformat(),
            processing_time=0.123,
        )


# Make TestHelpers available as a fixture