def METHOD_NAME(binary_mock_dataset_and_probabilities):
    # Arrange
    text_data, proba_train, proba_test = binary_mock_dataset_and_probabilities
    check = PredictionDrift()
    # Act
    result = check.run(text_data, text_data, train_probabilities=proba_train, test_probabilities=proba_test)
    # Assert
    assert_that(result.value['Drift score'], close_to(0.666, 0.01))