Skip to content

Commit a96b88d

Browse files
committed
Add code comments for tests
1 parent cf01e19 commit a96b88d

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

tests/tests/test_utils.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,15 +37,22 @@ def test_parent_template_from_variable(self):
3737
],
3838
)
3939

40+
41+
class TestGetTemplateDirs(SimpleTestCase):
42+
def get_relative_template_dirs(self):
43+
"""Make paths relative with a predefined root so we can use them in assertions."""
44+
base = os.path.dirname(settings.BASE_DIR)
45+
dirs = get_template_dirs()
46+
return ['/'.join(str(d).replace(base, 'dpl').split('/')[-4:-1]) for d in dirs]
47+
4048
@override_settings(TEMPLATES=[
4149
{
4250
'BACKEND': 'django.template.backends.django.DjangoTemplates',
4351
'APP_DIRS': True,
4452
},
4553
])
4654
def test_get_template_dirs_app_dirs(self):
47-
template_dirs = ['/'.join(d.replace(os.path.dirname(settings.BASE_DIR), 'dpl').split('/')[-4:-1]) for d in get_template_dirs()]
48-
self.assertListEqual(template_dirs, [
55+
self.assertListEqual(self.get_relative_template_dirs(), [
4956
'django/contrib/auth',
5057
'dpl/pattern_library',
5158
'dpl/tests',
@@ -65,8 +72,7 @@ def test_get_template_dirs_app_dirs(self):
6572
},
6673
])
6774
def test_get_template_dirs_list_dirs(self):
68-
template_dirs = ['/'.join(d.replace(os.path.dirname(settings.BASE_DIR), 'dpl').split('/')[-4:-1]) for d in get_template_dirs()]
69-
self.assertListEqual(template_dirs, [
75+
self.assertListEqual(self.get_relative_template_dirs(), [
7076
'dpl/tests/test_one',
7177
'dpl/tests/test_two',
7278
'django/contrib/auth',

0 commit comments

Comments
 (0)