Commit e244e456 authored by Aymeric Augustin's avatar Aymeric Augustin
Browse files

[1.7.x] Appeased flake8.

Backport of b1432bfc from master
parent 189bcb9b
Loading
Loading
Loading
Loading
+0 −1
Original line number Diff line number Diff line
@@ -3,7 +3,6 @@ import warnings
from django.db import models
from django.test import TestCase, override_settings
from django.utils import six
from django.core.files.storage import FileSystemStorage


class FieldDeconstructionTests(TestCase):
+2 −1
Original line number Diff line number Diff line
@@ -757,10 +757,12 @@ class OperationTests(MigrationTestBase):
        Tests the RunPython operation correctly handles the "atomic" keyword
        """
        project_state = self.set_up_test_model("test_runpythonatomic", mti_model=True)

        def inner_method(models, schema_editor):
            Pony = models.get_model("test_runpythonatomic", "Pony")
            Pony.objects.create(pink=1, weight=3.55)
            raise ValueError("Adrian hates ponies.")

        atomic_migration = Migration("test", "test_runpythonatomic")
        atomic_migration.operations = [migrations.RunPython(inner_method)]
        non_atomic_migration = Migration("test", "test_runpythonatomic")
@@ -789,7 +791,6 @@ class OperationTests(MigrationTestBase):
            self.assertEqual(project_state.render().get_model("test_runpythonatomic", "Pony").objects.count(), 1)



class MigrateNothingRouter(object):
    """
    A router that sends all writes to the other database.
+5 −1
Original line number Diff line number Diff line
@@ -236,14 +236,18 @@ class SchemaTests(TransactionTestCase):
        Tests adding fields to models with a default that is not directly
        valid in the database (#22581)
        """

        class TestTransformField(IntegerField):

            # Weird field that saves the count of items in its value
            def get_default(self):
                return self.default

            def get_prep_value(self, value):
                if value is None:
                    return 0
                return len(value)

        # Create the table
        with connection.schema_editor() as editor:
            editor.create_model(Author)
+3 −3

File changed.

Contains only whitespace changes.