diff options
Diffstat (limited to 'yaksh')
| -rw-r--r-- | yaksh/test_models.py | 4 | ||||
| -rw-r--r-- | yaksh/test_views.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/yaksh/test_models.py b/yaksh/test_models.py index c805c52..019e2fc 100644 --- a/yaksh/test_models.py +++ b/yaksh/test_models.py @@ -2611,14 +2611,14 @@ class QRcodeTestCase(unittest.TestCase): def test_generate_image(self): # Given qrcode = self.qrcode - image_name = 'qrcode/{0}.png'.format(qrcode.short_key) + image_name = qrcode.short_key # When qrcode.generate_image('test') # Then self.assertTrue(qrcode.is_active()) - self.assertEqual(qrcode.image.name, image_name) + self.assertIn(image_name, qrcode.image.name) def test_get_qrcode(self): # Given diff --git a/yaksh/test_views.py b/yaksh/test_views.py index 925c3d1..793f9c8 100644 --- a/yaksh/test_views.py +++ b/yaksh/test_views.py @@ -4352,7 +4352,7 @@ class TestSelfEnroll(TestCase): -class TestGrader(SimpleTestCase): +class TestGrader(TestCase): allow_database_queries = True def setUp(self): |