diff --git a/tests/test_settings.py b/tests/test_settings.py index 51cab51..f358557 100644 --- a/tests/test_settings.py +++ b/tests/test_settings.py @@ -28,7 +28,10 @@ class TestSettings(unittest.TestCase): self.assertEqual(self.settings['thumb_suffix'], '.tn') def test_thumb(self): - self.assertEqual(get_thumb(self.settings, 'example.jpg'), - 'thumbnails/example.tn.jpg') - self.assertEqual(get_thumb(self.settings, 'test/example.jpg'), - 'test/thumbnails/example.tn.jpg') + """Test the get_thumb function.""" + + tests = [('example.jpg', 'thumbnails/example.tn.jpg'), + ('test/example.jpg', 'test/thumbnails/example.tn.jpg'), + ('test/t/example.jpg', 'test/t/thumbnails/example.tn.jpg')] + for src, ref in tests: + self.assertEqual(get_thumb(self.settings, src), ref)