code: def test_prediction_ok_cases():
      data = {
          "multi_label": True,
          "inputs": {"data": "My cool data"},
          "prediction": {
              "agent": "test",
              "labels": [
                  {"class": "A", "score": 0.3},
                  {"class": "B", "score": 0.9},
              ],
          },
      }
  
      record = ServiceTextClassificationRecord(**data)
      assert record.predicted is None
      record.annotation = TextClassificationAnnotation(
          **{
              "agent": "test",
              "labels": [
                  {"class": "A", "score": 1},
                  {"class": "B", "score": 1},
              ],
          },
      )
      assert record.predicted == PredictionStatus.KO
  
      record.prediction = TextClassificationAnnotation(
          **{
              "agent": "test",
              "labels": [
                  {"class": "A", "score": 0.9},
                  {"class": "B", "score": 0.9},
              ],
          },
      )
      assert record.predicted == PredictionStatus.OK
  
      record.prediction = None
      assert record.predicted is None
  
imports: import pytest
  from argilla.server.apis.v0.models.text_classification import TextClassificationAnnotation, TextClassificationQuery, TextClassificationRecord
  from argilla.server.commons.models import PredictionStatus, TaskStatus
  from argilla.server.daos.backend.search.query_builder import EsQueryBuilder
  from argilla.server.services.tasks.text_classification.model import ClassPrediction, ServiceTextClassificationRecord
  from tests.pydantic_v1 import ValidationError
label_desc: Use image node_id 9 for calling a global function with example usage: test_prediction_ok_cases() without return types