From 310fd86c21f2ce0000d8023b95385d895c0f01bd Mon Sep 17 00:00:00 2001 From: Matthew Tretter Date: Fri, 17 Feb 2012 20:44:47 -0500 Subject: [PATCH] Extract utility functions --- tests/core/tests.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/tests/core/tests.py b/tests/core/tests.py index ba9605d..cf2b994 100644 --- a/tests/core/tests.py +++ b/tests/core/tests.py @@ -33,15 +33,19 @@ def generate_lenna(): return tmp -def create_photo(name): - photo = Photo() +def create_instance(model_class, image_name): + instance = model_class() img = generate_lenna() file = ContentFile(img.read()) - photo.original_image = file - photo.original_image.save(name, file) - photo.save() + instance.original_image = file + instance.original_image.save(image_name, file) + instance.save() img.close() - return photo + return instance + + +def create_photo(name): + return create_instance(Photo, name) class Photo(models.Model):