1  import pandas as pd
 2  
 3  from evidently.legacy.features.text_length_feature import TextLength
 4  from evidently.legacy.pipeline.column_mapping import ColumnMapping
 5  from evidently.legacy.utils.data_preprocessing import create_data_definition
 6  
 7  
 8  def test_text_length_feature():
 9      feature_generator = TextLength("column_1")
10      data = pd.DataFrame(dict(column_1=["abcdefg", "abc", "a"]))
11      result = feature_generator.generate_feature(
12          data=data,
13          data_definition=create_data_definition(None, data, ColumnMapping()),
14      )
15      assert result.equals(pd.DataFrame(dict(column_1=[7, 3, 1])))