def METHOD_NAME(small_wikiann_train_test_text_data):
    # Arrange
    train, test = small_wikiann_train_test_text_data
    train_label_with_nones = train.label
    train_label_with_nones[0][0] = None
    train = TextData(train.text, tokenized_text=train.tokenized_text,
                     task_type='token_classification')
    check = PredictionDrift()
    # Act
    result = check.run(train, test, train_predictions=np.asarray(train_label_with_nones),
                       test_predictions=np.asarray(test.label))
    # Assert
    assert_that(result.value['Drift score'], close_to(0, 0.04))