code: def test_regressor_categorical_variables(df_regression):
      X, y = df_regression
  
      tr = TargetMeanRegressor(variables="cat_var_A")
      tr.fit(X, y)
      pred = tr.predict(X)
  
      exp_pred = np.array(
          [
              0.0,
              0.0,
              0.0,
              0.0,
              0.0,
              1.0,
              1.0,
              1.0,
              1.0,
              1.0,
              2.0,
              2.0,
              2.0,
              2.0,
              2.0,
              3.0,
              3.0,
              3.0,
              3.0,
              3.0,
          ]
      )
  
      assert np.array_equal(pred, exp_pred)
  
      tr = TargetMeanRegressor(variables="cat_var_B")
      tr.fit(X, y)
  
      pred = tr.predict(X)
  
      exp_pred = np.array(
          [
              0.16666667,
              0.16666667,
              0.16666667,
              0.16666667,
              0.16666667,
              0.16666667,
              1.5,
              1.5,
              1.5,
              1.5,
              1.5,
              1.5,
              1.5,
              1.5,
              2.83333333,
              2.83333333,
              2.83333333,
              2.83333333,
              2.83333333,
              2.83333333,
          ]
      )
  
      assert np.allclose(pred, exp_pred)
  
      tr = TargetMeanRegressor(variables=["cat_var_A", "cat_var_B"])
      tr.fit(X, y)
  
      pred = tr.predict(X)
  
      exp_pred = np.array(
          [
              0.08333333,
              0.08333333,
              0.08333333,
              0.08333333,
              0.08333333,
              0.58333333,
              1.25,
              1.25,
              1.25,
              1.25,
              1.75,
              1.75,
              1.75,
              1.75,
              2.41666667,
              2.91666667,
              2.91666667,
              2.91666667,
              2.91666667,
              2.91666667,
          ]
      )
  
      assert np.allclose(pred, exp_pred)
  
imports: import numpy
  from feature_engine._prediction.target_mean_regressor import TargetMeanRegressor
label_desc: Use image node_id 1 for calling a global function with example usage: test_regressor_categorical_variables(df_regression) without return types