code: def test_error_if_input_df_contains_na_in_transform(
      df_vartypes, df_na
  ):
      # test case 1: when dataset contains na, transform method
      age_dict = {"Age": [0, 10, 20, 30, np.Inf]}
  
      with pytest.raises(ValueError):
          transformer = ArbitraryDiscretiser(binning_dict=age_dict)
          transformer.fit(df_vartypes)
          transformer.transform(
              df_na[["Name", "City", "Age", "Marks", "dob"]]
          )
  
file_path: feature_engine/tests/test_discretisation/test_arbitrary_discretiser.py
imports: import numpy
  import pandas
  import pytest
  from numpy.random import default_rng
  from scipy.stats import skewnorm
  from sklearn.datasets import fetch_california_housing
  from feature_engine.discretisation import ArbitraryDiscretiser
label_desc: Use image node_id 2 for calling a global function with example usage: test_error_if_input_df_contains_na_in_transform(df_vartypes, df_na) without return types