from metadata.sdk import configure
from metadata.sdk.entities import TestSuites

configure(
    host="https://your-company.open-metadata.org/api",
    jwt_token="your-jwt-token"
)

# Get by ID
ts = TestSuites.retrieve("e86a9a11-852f-4bac-b5a7-993b2bbbb572")
print(f"{ts.displayName} (basic: {ts.basic})")

# Get by ID with fields
ts = TestSuites.retrieve(
    "e86a9a11-852f-4bac-b5a7-993b2bbbb572",
    fields=["owners", "tests"]
)

# Get by fully qualified name
ts = TestSuites.retrieve_by_name("sample_data.ecommerce_db.shopify.dim_address.testSuite")

// client is an initialized OpenMetadataClient.
var testSuite = client.testSuites().get("e86a9a11-852f-4bac-b5a7-993b2bbbb572");

// Get by ID with fields
var testSuiteWithFields = client.testSuites().get(
    "e86a9a11-852f-4bac-b5a7-993b2bbbb572",
    "owners,tests");

// Get by fully qualified name
var testSuiteByName = client.testSuites().getByName("sample_data.ecommerce_db.shopify.dim_address.testSuite");

// Get by name with fields
var testSuiteByNameWithFields = client.testSuites().getByName(
    "sample_data.ecommerce_db.shopify.dim_address.testSuite",
    "owners,tests");

# Get by ID
curl "{base_url}/api/v1/dataQuality/testSuites/e86a9a11-852f-4bac-b5a7-993b2bbbb572" \
  -H "Authorization: Bearer {access_token}"

# Get by ID with fields
curl "{base_url}/api/v1/dataQuality/testSuites/e86a9a11-852f-4bac-b5a7-993b2bbbb572?fields=owners,tests" \
  -H "Authorization: Bearer {access_token}"

# Get by name
curl "{base_url}/api/v1/dataQuality/testSuites/name/sample_data.ecommerce_db.shopify.dim_address.testSuite" \
  -H "Authorization: Bearer {access_token}"

{
  "id": "e86a9a11-852f-4bac-b5a7-993b2bbbb572",
  "name": "sample_data.ecommerce_db.shopify.dim_address.testSuite",
  "displayName": "Data Contract - dim_address_comprehensive_contract",
  "fullyQualifiedName": "sample_data.ecommerce_db.shopify.dim_address.testSuite",
  "version": 0.1,
  "updatedAt": 1769982757893,
  "updatedBy": "admin",
  "deleted": false,
  "owners": [],
  "basic": true
}