code: def test_arg_probing(mock_handler_class):
      handler = mock_handler_class
  
      # Test create
      prediction_args = handler.prediction_args()
      # sort
      prediction_args = sorted(prediction_args, key=lambda x: x["name"])
  
      assert prediction_args == [
          {
              "name": "test_optional",
              "required": False,
          },
          {
              "name": "test_optional2",
              "required": False,
          },
          {
              "name": "test_required_at_some_point",
              "required": True,
          },
          {
              "name": "this_is_actually_required",
              "required": True,
          },
      ]
  
      creation_args = handler.creation_args()
      # sort
      creation_args = sorted(creation_args, key=lambda x: x["name"])
      assert creation_args == [
          {
              "name": "test_optional",
              "required": False,
          },
          {
              "name": "test_optional2",
              "required": False,
          },
          {
              "name": "test_required",
              "required": True,
          },
      ]
  
file_path: mindsdb/tests/unit/test_arg_probing.py
imports: import pytest
  from mindsdb.integrations.handlers.openai_handler.openai_handler import OpenAIHandler
  from mindsdb.integrations.handlers.statsforecast_handler.statsforecast_handler import StatsForecastHandler
  from mindsdb.integrations.libs.base import ArgProbeMixin
label_desc: Use image node_id 3 for calling a global function with example usage: test_arg_probing(mock_handler_class) without return types