Silenced a Python 2 ImportWarning in a migrations test.

This commit is contained in:
Tim Graham 2014-07-24 09:24:52 -04:00
parent 577ced2aae
commit 66211b4b75
1 changed files with 7 additions and 1 deletions

View File

@ -6,6 +6,7 @@ import os
import re import re
import tokenize import tokenize
import unittest import unittest
import warnings
from django.core.validators import RegexValidator, EmailValidator from django.core.validators import RegexValidator, EmailValidator
from django.db import models, migrations from django.db import models, migrations
@ -272,7 +273,12 @@ class WriterTests(TestCase):
migration = migrations.Migration('0001_initial', app.split('.')[-1]) migration = migrations.Migration('0001_initial', app.split('.')[-1])
expected_path = os.path.join(base_dir, *(app.split('.') + ['migrations', '0001_initial.py'])) expected_path = os.path.join(base_dir, *(app.split('.') + ['migrations', '0001_initial.py']))
writer = MigrationWriter(migration) writer = MigrationWriter(migration)
self.assertEqual(writer.path, expected_path) # Silence warning on Python 2: Not importing directory
# 'tests/migrations/migrations_test_apps/without_init_file/migrations':
# missing __init__.py
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=ImportWarning)
self.assertEqual(writer.path, expected_path)
def test_custom_operation(self): def test_custom_operation(self):
migration = type(str("Migration"), (migrations.Migration,), { migration = type(str("Migration"), (migrations.Migration,), {