def test_thumbnail_or_file(self): """thumbnail_or_file() returns self.thumbnail if set, or else returns self.file""" image = ImageAttachment(content_object=self.obj, creator=self.user) with open('apps/upload/tests/media/test.jpg') as f: up_file = File(f) image.file.save(up_file.name, up_file, save=True) eq_(image.file, image.thumbnail_or_file()) generate_thumbnail(image, up_file.name) eq_(image.thumbnail, image.thumbnail_or_file())