diff --git a/tests/commands/migrations/test_make_command.py b/tests/commands/migrations/test_make_command.py index b72269eb..af0c964d 100644 --- a/tests/commands/migrations/test_make_command.py +++ b/tests/commands/migrations/test_make_command.py @@ -31,7 +31,7 @@ def test_basic_create_gives_creator_proper_arguments_when_table_is_set(self): self.run_command(command, [("name", "create_foo"), ("--table", "users")]) def test_basic_create_gives_creator_proper_arguments_when_table_is_set_with_create( - self + self, ): creator_mock = flexmock(MigrationCreator) creator_mock.should_receive("create").once().with_args( diff --git a/tests/commands/test_command.py b/tests/commands/test_command.py index c1eda583..95fac783 100644 --- a/tests/commands/test_command.py +++ b/tests/commands/test_command.py @@ -13,7 +13,8 @@ class FooCommand(Command): """ Test Command """ - name = 'foo' + + name = "foo" def handle(self): pass @@ -21,14 +22,14 @@ def handle(self): class CommandTestCase(OratorCommandTestCase): def test_get_py_config_and_require___file__(self): - filename = tempfile.mktemp('.py') - with open(filename, 'w') as f: - f.write('foo = __file__') + filename = tempfile.mktemp(".py") + with open(filename, "w") as f: + f.write("foo = __file__") command = flexmock(FooCommand()) - command.should_call('_get_config').and_return({'foo': filename}) + command.should_call("_get_config").and_return({"foo": filename}) - self.run_command(command, [('-c', filename)]) + self.run_command(command, [("-c", filename)]) if os.path.exists(filename): os.remove(filename) diff --git a/tests/migrations/test_migrator.py b/tests/migrations/test_migrator.py index c62c3d2d..07141c97 100644 --- a/tests/migrations/test_migrator.py +++ b/tests/migrations/test_migrator.py @@ -224,7 +224,7 @@ def test_last_batch_of_migrations_can_be_rolled_back(self): migrator.rollback(os.getcwd()) def test_last_batch_of_migrations_can_be_rolled_back_directly_if_transactional_is_false( - self + self, ): resolver_mock = flexmock(DatabaseManager) resolver_mock.should_receive("connection").and_return({})