code: def test_with_drift_classification(
      mnist_visiondata_train, mnist_visiondata_test
  ):
      # Arrange
      check = PredictionDrift(categorical_drift_method="PSI")
  
      # Act
      result = check.run(mnist_visiondata_train, mnist_visiondata_test)
  
      # Assert
      assert_that(
          result.value,
          has_entries(
              {
                  "Samples Per Class": has_entries(
                      {
                          "Drift score": close_to(0.057, 0.01),
                          "Method": equal_to("PSI"),
                      }
                  )
              }
          ),
      )
  
file_path: deepchecks/tests/vision/checks/model_evaluation/prediction_drift_test.py
imports: from hamcrest import assert_that, calling, close_to, equal_to, greater_than, has_entries, has_length, raises
  from deepchecks.core.errors import NotEnoughSamplesError
  from deepchecks.vision.checks import PredictionDrift
  from tests.base.utils import equal_condition_result
label_desc: Use image node_id 4 for calling a global function with example usage: test_with_drift_classification(mnist_visiondata_train, mnist_visiondata_test) without return types