From ccaafe492c1a85398f67ad5cf4d912e99f36c7b5 Mon Sep 17 00:00:00 2001
From: Jannis <jvajen@gmail.com>
Date: Thu, 2 Feb 2017 11:28:09 +0100
Subject: [PATCH] Fix test for avatar tag.

---
 tests/tests.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/tests/tests.py b/tests/tests.py
index b1763f7..8ae0b98 100644
--- a/tests/tests.py
+++ b/tests/tests.py
@@ -215,9 +215,8 @@ class AvatarTests(TestCase):
         avatar = get_primary_avatar(self.user)
 
         result = avatar_tags.avatar(self.user, title="Avatar")
-
-        self.assertIn('<img src="{}"'.format(avatar.avatar_url(80)), result)
-        self.assertIn('width="80" height="80" alt="test" title="Avatar" />', result)
+        html = '<img src="{}" width="80" height="80" alt="test" title="Avatar" />'.format(avatar.avatar_url(80))
+        self.assertInHTML(html, result)
 
     def test_default_add_template(self):
         response = self.client.get('/avatar/add/')
@@ -251,7 +250,7 @@ class AvatarTests(TestCase):
         response = self.client.get('/avatar/delete/')
         self.assertNotContains(response, 'like to delete.')
         self.assertContains(response, 'ALTERNATE DELETE TEMPLATE')
-    
+
     # def testAvatarOrder
     # def testReplaceAvatarWhenMaxIsOne
     # def testHashFileName