Switching this test up a little

theonion · Apr 15, 2014 · 3168782 · 3168782
1 parent 17df2dc
commit 3168782
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_image_field.py b/tests/test_image_field.py
@@ -57,9 +57,11 @@ def setUp(self):
 
     def test_fileless_save(self):
         test_object = TestModel()
-        test_object.listing_image.id = 12345
+        test_object.listing_image = 12345
         test_object.save()
 
+        self.assertEqual(test_object.listing_image.id, 12345)
+
     def test_save(self):
         lenna_path = os.path.join(TEST_DATA_PATH, 'Lenna.png')
         with open(lenna_path, "rb") as lenna: