code: def test_athena_query_run(sagemaker_session_mock, query):
      sagemaker_session_mock.start_query_execution.return_value = {
          "QueryExecutionId": "query_id"
      }
      query.run(
          query_string="query",
          output_location="s3://some-bucket/some-path",
          workgroup="workgroup",
      )
      sagemaker_session_mock.start_query_execution.assert_called_with(
          catalog="catalog",
          database="database",
          query_string="query",
          output_location="s3://some-bucket/some-path",
          kms_key=None,
          workgroup="workgroup",
      )
      assert "some-bucket" == query._result_bucket
      assert "some-path" == query._result_file_prefix
      assert "query_id" == query._current_query_execution_id
  
imports: from __future__ import absolute_import
  import pandas
  import pytest
  from mock import Mock, patch, MagicMock
  from botocore.exceptions import ProfileNotFound
  from sagemaker.feature_store.feature_definition import FractionalFeatureDefinition, IntegralFeatureDefinition, StringFeatureDefinition, FeatureTypeEnum, VectorCollectionType, SetCollectionType, ListCollectionType
  from sagemaker.feature_store.feature_group import FeatureGroup, IngestionManagerPandas, AthenaQuery, IngestionError
  from sagemaker.feature_store.inputs import FeatureParameter, DeletionModeEnum, TtlDuration, OnlineStoreConfigUpdate, OnlineStoreStorageTypeEnum
  from tests.unit import SAGEMAKER_CONFIG_FEATURE_GROUP
label_desc: Use image node_id 43 for calling a global function with example usage: test_athena_query_run(sagemaker_session_mock, query) without return types