code: def test_databricks_command_run_context_tags():
      with mock.patch(
          "mlflow.utils.databricks_utils.get_job_group_id"
      ) as job_group_id_mock:
          assert DatabricksCommandRunContext().tags() == {
              MLFLOW_DATABRICKS_NOTEBOOK_COMMAND_ID: job_group_id_mock.return_value
          }
  
imports: from unittest import mock
  from mlflow.tracking.context.databricks_command_context import DatabricksCommandRunContext
  from mlflow.utils.mlflow_tags import MLFLOW_DATABRICKS_NOTEBOOK_COMMAND_ID
label_desc: Use image node_id 2 for calling a global function with example usage: test_databricks_command_run_context_tags() without return types