code: def test_create_collection_success(sagemaker_session):
      collection = Collection(sagemaker_session)
      create_response = collection.create(
          collection_name=COLLECTION_NAME
      )
      assert create_response["Name"] is COLLECTION_NAME
      assert create_response["Arn"] is not None
  
file_path: sagemaker-python-sdk/tests/unit/test_collection.py
imports: from __future__ import absolute_import
  import pytest
  import json
  from mock import Mock
  from sagemaker.collection import Collection
label_desc: Use image node_id 2 for calling a global function with example usage: test_create_collection_success(sagemaker_session) without return types