code: def test_find_or_check_numerical_variables(
      df_vartypes, df_numeric_columns
  ):
      vars_num = ["Age", "Marks"]
      var_num = "Age"
      vars_mix = ["Age", "Marks", "Name"]
      vars_none = None
  
      assert (
          _find_or_check_numerical_variables(df_vartypes, vars_num)
          == vars_num
      )
      assert _find_or_check_numerical_variables(
          df_vartypes, var_num
      ) == ["Age"]
      assert (
          _find_or_check_numerical_variables(df_vartypes, vars_none)
          == vars_num
      )
  
      with pytest.raises(TypeError):
          assert _find_or_check_numerical_variables(df_vartypes, "City")
  
      with pytest.raises(TypeError):
          assert _find_or_check_numerical_variables(
              df_numeric_columns, 0
          )
  
      with pytest.raises(TypeError):
          assert _find_or_check_numerical_variables(
              df_numeric_columns, [1, 3]
          )
  
      with pytest.raises(TypeError):
          assert _find_or_check_numerical_variables(
              df_vartypes, vars_mix
          )
  
      with pytest.raises(ValueError):
          assert _find_or_check_numerical_variables(
              df_vartypes, variables=[]
          )
  
      with pytest.raises(ValueError):
          assert _find_or_check_numerical_variables(
              df_vartypes[["Name", "City"]], None
          )
  
      assert _find_or_check_numerical_variables(
          df_numeric_columns, [2, 3]
      ) == [2, 3]
      assert _find_or_check_numerical_variables(
          df_numeric_columns, 2
      ) == [2]
  
file_path: feature_engine/tests/test_variable_handling/test_variable_type_selection.py
imports: import pandas
  import pytest
  from feature_engine._variable_handling.variable_type_selection import _filter_out_variables_not_in_dataframe, _find_all_variables, _find_categorical_and_numerical_variables, _find_or_check_categorical_variables, _find_or_check_datetime_variables, _find_or_check_numerical_variables
label_desc: Use image node_id 1 for calling a global function with example usage: test_find_or_check_numerical_variables(df_vartypes, df_numeric_columns) without return types