1  from evidently.legacy.core import ColumnType
 2  from evidently.legacy.test_preset import DataDriftTestPreset
 3  from evidently.legacy.tests import TestAllFeaturesValueDrift
 4  from evidently.legacy.tests import TestEmbeddingsDrift
 5  from evidently.legacy.tests import TestShareOfDriftedColumns
 6  from evidently.legacy.utils.data_preprocessing import ColumnDefinition
 7  from evidently.legacy.utils.data_preprocessing import DataDefinition
 8  
 9  
10  def test_embeddings_data_drift_preset():
11      data_definition = DataDefinition(
12          columns={
13              "target": ColumnDefinition("target", ColumnType.Numerical),
14          },
15          embeddings={
16              "small_set": ["col_1", "col_2"],
17              "big_set": ["col_3", "col_4"],
18          },
19          reference_present=True,
20      )
21      preset = DataDriftTestPreset(embeddings=["small_set", "big_set"])
22      tests = preset.generate_tests(data_definition=data_definition, additional_data=None)
23      assert len(tests) == 4
24      expected_tests = [TestShareOfDriftedColumns, TestAllFeaturesValueDrift, TestEmbeddingsDrift, TestEmbeddingsDrift]
25      assert expected_tests == [type(test) for test in tests]