test(sqlalchemy-spanner): add mockserver 15-matrix tests and sample for types.Uuid vs types.UUID

Author: sakthivelmanii

packages/sqlalchemy-spanner/samples/uuid_sample.py

@@ -54,9 +54,7 @@ class CustomerLegacy(Base):
     """Uses types.Uuid() -> Compiles to STRING(36) by default."""
 
     __tablename__ = "customers_legacy"
-    id: Mapped[UUID] = mapped_column(
-        types.Uuid(), primary_key=True, default=uuid.uuid4
-    )
+    id: Mapped[UUID] = mapped_column(types.Uuid(), primary_key=True, default=uuid.uuid4)
     name: Mapped[str] = mapped_column()
 
 
@@ -67,9 +65,7 @@ class CustomerNative(Base):
     """
 
     __tablename__ = "customers_native"
-    id: Mapped[UUID] = mapped_column(
-        types.UUID(), primary_key=True, default=uuid.uuid4
-    )
+    id: Mapped[UUID] = mapped_column(types.UUID(), primary_key=True, default=uuid.uuid4)
     name: Mapped[str] = mapped_column()
 
 
@@ -78,7 +74,9 @@ class CustomerExplicitString(Base):
 
     __tablename__ = "customers_explicit_string"
     id: Mapped[UUID] = mapped_column(
-        types.Uuid(native_uuid=False), primary_key=True, default=uuid.uuid4
+        types.Uuid(native_uuid=False),
+        primary_key=True,
+        default=uuid.uuid4,
     )
     name: Mapped[str] = mapped_column()
 
@@ -104,9 +102,7 @@ def run_with_flag_disabled():
 
     with Session(engine) as session:
         cust_legacy = CustomerLegacy(name="Alice_Legacy_FlagDisabled")
-        cust_override = CustomerExplicitString(
-            name="Diana_String_FlagDisabled"
-        )
+        cust_override = CustomerExplicitString(name="Diana_String_FlagDisabled")
         session.add_all([cust_legacy, cust_override])
         session.commit()
 

packages/sqlalchemy-spanner/tests/mockserver_tests/test_uuid.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import uuid
+from uuid import UUID
 
 from google.cloud.spanner_v1 import TypeCode
 from sqlalchemy import Column, MetaData, Table, select, types
@@ -37,18 +38,19 @@ class Base(DeclarativeBase):
 
 class UserUuid(Base):
     __tablename__ = "users_uuid"
-    id: Mapped[uuid.UUID] = mapped_column(types.Uuid(), primary_key=True)
+    id: Mapped[UUID] = mapped_column(types.Uuid(), primary_key=True)
 
 
 class UserUUID(Base):
     __tablename__ = "users_UUID"
-    id: Mapped[uuid.UUID] = mapped_column(types.UUID(), primary_key=True)
+    id: Mapped[UUID] = mapped_column(types.UUID(), primary_key=True)
 
 
 class UserUuidNativeFalse(Base):
     __tablename__ = "users_uuid_native_false"
-    id: Mapped[uuid.UUID] = mapped_column(
-        types.Uuid(native_uuid=False), primary_key=True
+    id: Mapped[UUID] = mapped_column(
+        types.Uuid(native_uuid=False),
+        primary_key=True,
     )