fix: ruff format and remove unused imports in serde contract tests

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: cmgrote

pyatlan_v9/model/typedef.py

@@ -463,7 +463,9 @@ class Options(msgspec.Struct, kw_only=True, rename="camel", omit_defaults=True):
         applicable_entity_types: Union[str, None, msgspec.UnsetType] = msgspec.UNSET
         """Set of entities on which this attribute can be applied (JSON-encoded)."""
 
-        custom_applicable_entity_types: Union[str, None, msgspec.UnsetType] = msgspec.UNSET
+        custom_applicable_entity_types: Union[str, None, msgspec.UnsetType] = (
+            msgspec.UNSET
+        )
         """Deprecated: see applicable_asset_types, applicable_glossary_types."""
 
         allow_search: Union[bool, None, msgspec.UnsetType] = msgspec.UNSET
@@ -543,8 +545,8 @@ class Options(msgspec.Struct, kw_only=True, rename="camel", omit_defaults=True):
         )
         """AI asset type names to restrict the attribute (JSON-encoded)."""
 
-        applicable_other_asset_types: Union[str, None, msgspec.UnsetType] = msgspec.field(
-            default=msgspec.UNSET, name="otherAssetTypeList"
+        applicable_other_asset_types: Union[str, None, msgspec.UnsetType] = (
+            msgspec.field(default=msgspec.UNSET, name="otherAssetTypeList")
         )
         """Other asset type names to restrict the attribute (JSON-encoded)."""
 
@@ -661,13 +663,13 @@ def creator(
     values_max_count: Union[float, None, msgspec.UnsetType] = msgspec.UNSET
     """Maximum number of values for this attribute."""
 
-    index_type_es_config: Union[Dict[str, Any], None, msgspec.UnsetType] = msgspec.field(
-        default=msgspec.UNSET, name="indexTypeESConfig"
+    index_type_es_config: Union[Dict[str, Any], None, msgspec.UnsetType] = (
+        msgspec.field(default=msgspec.UNSET, name="indexTypeESConfig")
     )
     """Internal use only."""
 
-    index_type_es_fields: Union[Dict[str, Dict[str, str]], None, msgspec.UnsetType] = msgspec.field(
-        default=msgspec.UNSET, name="indexTypeESFields"
+    index_type_es_fields: Union[Dict[str, Dict[str, str]], None, msgspec.UnsetType] = (
+        msgspec.field(default=msgspec.UNSET, name="indexTypeESFields")
     )
     """Internal use only."""
 
@@ -686,7 +688,11 @@ def __post_init__(self):
 
     def _get_option_set(self, attr: str) -> Set[str]:
         """Helper to parse a JSON-encoded set from options."""
-        val = getattr(self.options, attr, None) if self.options and self.options is not msgspec.UNSET else None
+        val = (
+            getattr(self.options, attr, None)
+            if self.options and self.options is not msgspec.UNSET
+            else None
+        )
         if val:
             return set(json.loads(val))
         return set()
@@ -1261,7 +1267,9 @@ class AtlanTagDef(TypeDef, kw_only=True):
     service_type: Union[str, None, msgspec.UnsetType] = msgspec.UNSET
     """Name used for display purposes."""
 
-    skip_display_name_uniqueness_check: Union[bool, None, msgspec.UnsetType] = msgspec.UNSET
+    skip_display_name_uniqueness_check: Union[bool, None, msgspec.UnsetType] = (
+        msgspec.UNSET
+    )
     """TBC"""
 
     @staticmethod
@@ -1324,13 +1332,17 @@ class EntityDef(TypeDef, kw_only=True):
     attribute_defs: Union[List[AttributeDef], msgspec.UnsetType] = msgspec.UNSET
     """Unused."""
 
-    business_attribute_defs: Union[Dict[str, List[Dict[str, Any]]], None, msgspec.UnsetType] = msgspec.UNSET
+    business_attribute_defs: Union[
+        Dict[str, List[Dict[str, Any]]], None, msgspec.UnsetType
+    ] = msgspec.UNSET
     """Unused."""
 
     category: AtlanTypeCategory = AtlanTypeCategory.ENTITY
     """Type category for entity definitions."""
 
-    relationship_attribute_defs: Union[List[RelationshipAttributeDef], msgspec.UnsetType] = msgspec.UNSET
+    relationship_attribute_defs: Union[
+        List[RelationshipAttributeDef], msgspec.UnsetType
+    ] = msgspec.UNSET
     """Unused."""
 
     service_type: Union[str, None, msgspec.UnsetType] = msgspec.UNSET
@@ -1353,7 +1365,9 @@ def reserved_type(self) -> bool:
 # =============================================================================
 
 
-class RelationshipEndDef(msgspec.Struct, kw_only=True, rename="camel", omit_defaults=True):
+class RelationshipEndDef(
+    msgspec.Struct, kw_only=True, rename="camel", omit_defaults=True
+):
     """Defines the structure of one end of a relationship.
 
     Despite Java's ``RelationshipEndDef extends AttributeDef``, analysis of
@@ -1406,7 +1420,9 @@ class RelationshipDef(TypeDef, kw_only=True):
     propagate_tags: Union[str, None, msgspec.UnsetType] = msgspec.UNSET
     """Whether Atlan tags propagate through this relationship, and in which direction."""
 
-    relationship_attribute_defs: Union[List[RelationshipAttributeDef], msgspec.UnsetType] = msgspec.UNSET
+    relationship_attribute_defs: Union[
+        List[RelationshipAttributeDef], msgspec.UnsetType
+    ] = msgspec.UNSET
     """Relationship attributes that can exist for this relationship type."""
 
     relationship_category: Union[str, None, msgspec.UnsetType] = msgspec.UNSET

tests_v9/unit/test_v9_typedef_serde_contract.py

@@ -28,19 +28,15 @@
 from __future__ import annotations
 
 import json
-from typing import Any, Dict, List
+from typing import Any, Dict
 
 import msgspec
 import pytest
 
 from pyatlan_v9.model.typedef import (
     AttributeDef,
-    CustomMetadataDef,
-    EntityDef,
     EnumDef,
-    RelationshipDef,
     RelationshipEndDef,
-    StructDef,
     TypeDefResponse,
 )
 
@@ -188,7 +184,9 @@ def test_sparse_attribute_def_does_not_expand_on_reencode(self) -> None:
         """
         sparse_source = {"name": "x", "typeName": "string"}
         full_payload = _payload(
-            entityDefs=[{"name": "E", "typeVersion": "1.0", "attributeDefs": [sparse_source]}]
+            entityDefs=[
+                {"name": "E", "typeVersion": "1.0", "attributeDefs": [sparse_source]}
+            ]
         )
         decoded = _decode(full_payload)
         re_encoded_attr = _encode(decoded)["entityDefs"][0]["attributeDefs"][0]
@@ -378,7 +376,9 @@ def test_explicit_relationship_category_survives(self) -> None:
         )
         decoded = _decode(raw)
         re_encoded = _encode(decoded)
-        assert re_encoded["relationshipDefs"][0]["relationshipCategory"] == "COMPOSITION"
+        assert (
+            re_encoded["relationshipDefs"][0]["relationshipCategory"] == "COMPOSITION"
+        )
 
 
 # ---------------------------------------------------------------------------
@@ -791,7 +791,11 @@ def test_relationship_attribute_defs_present_decode_as_structs(self) -> None:
                     "endDef1": {"type": "A", "name": "r1", "isContainer": False},
                     "endDef2": {"type": "B", "name": "r2", "isContainer": False},
                     "relationshipAttributeDefs": [
-                        {"name": "label", "typeName": "string", "relationshipTypeName": "R"}
+                        {
+                            "name": "label",
+                            "typeName": "string",
+                            "relationshipTypeName": "R",
+                        }
                     ],
                 }
             ]