1  import pandas as pd
 2  
 3  from evidently.legacy.test_preset import DataStabilityTestPreset
 4  from evidently.legacy.test_suite import TestSuite
 5  
 6  
 7  def test_data_stability_preset():
 8      test_current_dataset = pd.DataFrame(
 9          {
10              "category_feature": ["t", "e", "", ""],
11              "numerical_feature": [0.4, -12, None, 234],
12              "target": [1, 1, None, 1],
13              "prediction": [0, 0, None, 0],
14          }
15      )
16      test_reference_dataset = pd.DataFrame(
17          {
18              "category_feature": ["y", "n", "n", "y"],
19              "numerical_feature": [0, 1, 2, 5],
20              "target": [0, 0, 0, 1],
21              "prediction": [0, 0, 0, 1],
22          }
23      )
24      data_quality_suite = TestSuite(
25          tests=[
26              DataStabilityTestPreset(),
27          ]
28      )
29  
30      data_quality_suite.run(current_data=test_current_dataset, reference_data=test_reference_dataset)
31      assert not data_quality_suite
32      assert len(data_quality_suite.as_dict()["tests"]) == 10