fixing tests

Author: dantownsend

tests/apps/migrations/auto/test_diffable_table.py

@@ -1,9 +1,8 @@
-
 from unittest import TestCase
 
 from piccolo.apps.migrations.auto.diffable_table import (
     compare_dicts,
-    DiffableTable
+    DiffableTable,
 )
 from piccolo.columns import Varchar
 
@@ -18,30 +17,17 @@ def test_compare_dicts(self):
 
 class TestDiffableTable(TestCase):
     def test_subtract(self):
-        kwargs = {
-            'class_name': "Manager",
-            'tablename': 'manager'
-        }
+        kwargs = {"class_name": "Manager", "tablename": "manager"}
 
         name_column_1 = Varchar(unique=False)
         name_column_1._meta.name = "name"
-        table_1 = DiffableTable(
-            **kwargs,
-            columns=[name_column_1]
-        )
+        table_1 = DiffableTable(**kwargs, columns=[name_column_1])
 
         name_column_2 = Varchar(unique=True)
         name_column_2._meta.name = "name"
-        table_2 = DiffableTable(
-            **kwargs,
-            columns=[name_column_2]
-        )
-
-        delta = table_1 - table_2
-
-        self.assertEqual(
-            delta.alter_columns[0].params, {'unique': True}
-        )
-        self.assertEqual(
-            delta.alter_columns[0].old_params, {'unique': False}
-        )
+        table_2 = DiffableTable(**kwargs, columns=[name_column_2])
+
+        delta = table_2 - table_1
+
+        self.assertEqual(delta.alter_columns[0].params, {"unique": True})
+        self.assertEqual(delta.alter_columns[0].old_params, {"unique": False})

tests/apps/migrations/auto/test_schema_differ.py

@@ -174,7 +174,7 @@ def test_alter_column_precision(self):
         self.assertTrue(len(schema_differ.alter_columns.statements) == 1)
         self.assertEqual(
             schema_differ.alter_columns.statements[0],
-            "manager.alter_column(table_class_name='Ticket', tablename='ticket', column_name='price', params={'digits': (5, 2)}, old_params={'digits': (4, 2)})",  # noqa
+            "manager.alter_column(table_class_name='Ticket', tablename='ticket', column_name='price', params={'digits': (4, 2)}, old_params={'digits': (5, 2)})",  # noqa
         )
 
     def test_alter_default(self):

tests/coverage-report.sh

@@ -1,2 +1,2 @@
 #!/bin/bash
-./run-tests.sh --cov-report html $@
+./test-postgres.sh --cov-report html $@