diff options
| -rwxr-xr-x | yoyo/migrations.py | 4 | ||||
| -rw-r--r-- | yoyo/tests/__init__.py | 1 |
2 files changed, 3 insertions, 2 deletions
diff --git a/yoyo/migrations.py b/yoyo/migrations.py index 9ade1ae..b07f0af 100755 --- a/yoyo/migrations.py +++ b/yoyo/migrations.py @@ -244,8 +244,8 @@ class MigrationStep(StepBase): format = '|'.join(' %%- %ds ' % size for size in column_sizes) format += '\n' out.write(format % tuple(column_names)) - out.write('+'.join('-' * (size + 2) for size in column_sizes) - + '\n') + out.write('+'.join('-' * (size + 2) for size in column_sizes) + + '\n') for row in result: out.write(format % tuple(row)) out.write(plural(len(result), '(%d row)', '(%d rows)') + "\n") diff --git a/yoyo/tests/__init__.py b/yoyo/tests/__init__.py index 03021cb..9561c27 100644 --- a/yoyo/tests/__init__.py +++ b/yoyo/tests/__init__.py @@ -71,5 +71,6 @@ class MigrationsContextManager(object): return func(*(args + (self.tmpdir,)), **kwargs) return decorator + with_migrations = MigrationsContextManager migrations_dir = MigrationsContextManager |