1  import nltk
 2  import pandas as pd
 3  
 4  from evidently.legacy.features.trigger_words_presence_feature import TriggerWordsPresent
 5  from evidently.legacy.pipeline.column_mapping import ColumnMapping
 6  from evidently.legacy.utils.data_preprocessing import create_data_definition
 7  
 8  nltk.download("words")
 9  nltk.download("wordnet")
10  nltk.download("omw-1.4")
11  
12  
13  def test_trigger_words_present_feature():
14      feature_generator = TriggerWordsPresent("column_1", words_list=("apple",))
15      data = pd.DataFrame(dict(column_1=["Who are you and where are my apples?", "abc ", "a"]))
16      result = feature_generator.generate_feature(
17          data=data,
18          data_definition=create_data_definition(None, data, ColumnMapping()),
19      )
20      assert result.equals(pd.DataFrame(dict(column_1_apple_True=[1, 0, 0])))