From 7db7fd869a6bb617f767fad5b71ddf7cb5f14ff5 Mon Sep 17 00:00:00 2001
From: Kai Mueller <15907922+kasium@users.noreply.github.com>
Date: Wed, 1 Dec 2021 15:17:41 +0000
Subject: [PATCH] Fix tests

---
 test/orm/declarative/test_typing_py3k.py | 24 +++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/test/orm/declarative/test_typing_py3k.py b/test/orm/declarative/test_typing_py3k.py
index 7778e7c0fe..823fe54f10 100644
--- a/test/orm/declarative/test_typing_py3k.py
+++ b/test/orm/declarative/test_typing_py3k.py
@@ -5,19 +5,21 @@ from typing import TypeVar
 from sqlalchemy import Column
 from sqlalchemy import Integer
 from sqlalchemy.orm import as_declarative
+from sqlalchemy.testing import fixtures
 
 
-def test_class_getitem():
-    T = TypeVar("T", bound="CommonBase")  # noqa
+class DeclarativeBaseTest(fixtures.TestBase):
+    def test_class_getitem(self):
+        T = TypeVar("T", bound="CommonBase")  # noqa
 
-    class CommonBase(Generic[T]):
-        @classmethod
-        def boring(cls: Type[T]) -> Type[T]:
-            return cls
+        class CommonBase(Generic[T]):
+            @classmethod
+            def boring(cls: Type[T]) -> Type[T]:
+                return cls
 
-    @as_declarative()
-    class Base(CommonBase[T]):
-        pass
+        @as_declarative()
+        class Base(CommonBase[T]):
+            pass
 
-    class Tab(Base["Tab"]):
-        a = Column(Integer, primary_key=True)
+        class Tab(Base["Tab"]):
+            a = Column(Integer, primary_key=True)
-- 
2.47.3