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

tests/columns/test_timestamp.py

@@ -1,9 +1,9 @@
 import datetime
-from unittest import TestCase
 
 from piccolo.columns.column_types import Timestamp
 from piccolo.columns.defaults.timestamp import TimestampNow
 from piccolo.table import Table
+from tests.base import TableTest
 
 
 class MyTable(Table):
@@ -19,12 +19,8 @@ class MyTableDefault(Table):
     created_on = Timestamp(default=TimestampNow())
 
 
-class TestTimestamp(TestCase):
-    def setUp(self):
-        MyTable.create_table().run_sync()
-
-    def tearDown(self):
-        MyTable.alter().drop_table().run_sync()
+class TestTimestamp(TableTest):
+    tables = [MyTable]
 
     def test_timestamp(self):
         """
@@ -46,12 +42,8 @@ class TestTimestamp(TestCase):
             Timestamp(default=datetime.datetime.now(tz=datetime.timezone.utc))
 
 
-class TestTimestampDefault(TestCase):
-    def setUp(self):
-        MyTableDefault.create_table().run_sync()
-
-    def tearDown(self):
-        MyTableDefault.alter().drop_table().run_sync()
+class TestTimestampDefault(TableTest):
+    tables = [MyTableDefault]
 
     def test_timestamp(self):
         """

tests/columns/test_timestamptz.py

@@ -1,5 +1,4 @@
 import datetime
-from unittest import TestCase
 
 from dateutil import tz
 
@@ -10,6 +9,7 @@ from piccolo.columns.defaults.timestamptz import (
     TimestamptzOffset,
 )
 from piccolo.table import Table
+from tests.base import TableTest
 
 
 class MyTable(Table):
@@ -34,12 +34,8 @@ class CustomTimezone(datetime.tzinfo):
     pass
 
 
-class TestTimestamptz(TestCase):
-    def setUp(self):
-        MyTable.create_table().run_sync()
-
-    def tearDown(self):
-        MyTable.alter().drop_table().run_sync()
+class TestTimestamptz(TableTest):
+    tables = [MyTable]
 
     def test_timestamptz_timezone_aware(self):
         """
@@ -78,12 +74,8 @@ class TestTimestamptz(TestCase):
             self.assertEqual(result.created_on.tzinfo, datetime.timezone.utc)
 
 
-class TestTimestamptzDefault(TestCase):
-    def setUp(self):
-        MyTableDefault.create_table().run_sync()
-
-    def tearDown(self):
-        MyTableDefault.alter().drop_table().run_sync()
+class TestTimestamptzDefault(TableTest):
+    tables = [MyTableDefault]
 
     def test_timestamptz_default(self):
         """

tests/columns/test_uuid.py

@@ -1,20 +1,16 @@
 import uuid
-from unittest import TestCase
 
 from piccolo.columns.column_types import UUID
 from piccolo.table import Table
+from tests.base import TableTest
 
 
 class MyTable(Table):
     uuid = UUID()
 
 
-class TestUUID(TestCase):
-    def setUp(self):
-        MyTable.create_table().run_sync()
-
-    def tearDown(self):
-        MyTable.alter().drop_table().run_sync()
+class TestUUID(TableTest):
+    tables = [MyTable]
 
     def test_return_type(self):
         row = MyTable()

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]