def METHOD_NAME():
    records = [TextClassificationRecord(text="test", annotation="mock", prediction=[("mock2", 0.1)])]
    with pytest.raises(
        MissingPredictionError,
        match="It seems predictions are missing for the label 'mock'",
    ):
        _construct_s_and_psx(records)
    records.append(TextClassificationRecord(text="test", annotation="mock", prediction=[("mock", 0.1)]))
    with pytest.raises(
        MissingPredictionError,
        match="It seems a prediction for 'mock' is missing in the following record",
    ):
        _construct_s_and_psx(records)