code: def query(sagemaker_session_mock):
      return AthenaQuery(
          catalog="catalog",
          database="database",
          table_name="table_name",
          sagemaker_session=sagemaker_session_mock,
      )
  
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 42 for calling a global function with example usage: query(sagemaker_session_mock) and returns: AthenaQuery