piccolo 1.11.0__py3-none-any.whl → 1.12.0__py3-none-any.whl

tests/columns/test_varchar.py

@@ -1,9 +1,7 @@
-from unittest import TestCase
-
 from piccolo.columns.column_types import Varchar
 from piccolo.table import Table
 
-from ..base import engines_only
+from ..base import TableTest, engines_only
 
 
 class MyTable(Table):
@@ -11,7 +9,7 @@ class MyTable(Table):
 
 
 @engines_only("postgres", "cockroach")
-class TestVarchar(TestCase):
+class TestVarchar(TableTest):
     """
     SQLite doesn't enforce any constraints on max character length.
 
@@ -20,11 +18,7 @@ class TestVarchar(TestCase):
     Might consider enforcing this at the ORM level instead in the future.
     """
 
-    def setUp(self):
-        MyTable.create_table().run_sync()
-
-    def tearDown(self):
-        MyTable.alter().drop_table().run_sync()
+    tables = [MyTable]
 
     def test_length(self):
         row = MyTable(name="bob")

tests/query/functions/base.py

@@ -1,21 +1,8 @@
-import typing as t
-from unittest import TestCase
-
-from piccolo.table import Table, create_db_tables_sync, drop_db_tables_sync
+from tests.base import TableTest
 from tests.example_apps.music.tables import Band, Manager
 
 
-class FunctionTest(TestCase):
-    tables: t.List[t.Type[Table]]
-
-    def setUp(self) -> None:
-        create_db_tables_sync(*self.tables)
-
-    def tearDown(self) -> None:
-        drop_db_tables_sync(*self.tables)
-
-
-class BandTest(FunctionTest):
+class BandTest(TableTest):
     tables = [Band, Manager]
 
     def setUp(self) -> None:

tests/query/functions/test_datetime.py

@@ -12,16 +12,14 @@ from piccolo.query.functions.datetime import (
     Year,
 )
 from piccolo.table import Table
-from tests.base import engines_only, sqlite_only
-
-from .base import FunctionTest
+from tests.base import TableTest, engines_only, sqlite_only
 
 
 class Concert(Table):
     starts = Timestamp()
 
 
-class DatetimeTest(FunctionTest):
+class DatetimeTest(TableTest):
     tables = [Concert]
 
     def setUp(self) -> None:

tests/query/functions/test_math.py

@@ -3,15 +3,14 @@ import decimal
 from piccolo.columns import Numeric
 from piccolo.query.functions.math import Abs, Ceil, Floor, Round
 from piccolo.table import Table
-
-from .base import FunctionTest
+from tests.base import TableTest
 
 
 class Ticket(Table):
     price = Numeric(digits=(5, 2))
 
 
-class TestMath(FunctionTest):
+class TestMath(TableTest):
 
     tables = [Ticket]