def test_image_creation_with_user(self):
print '- test_image_creation_with_user'
image_file = FileStorage(
stream=open('fast_style_transfer/source-images/cape-flattery.jpg'))
user = User.query.get(1)
image = Image.create(image_file, user)
self.assertEqual(image.user_id, 1)
self.assertIsInstance(image.user, User)
self.assertEqual(image.is_public, user.pref_is_public)
print '+ passed'
def test_image_creation_no_user(self):
print '- test_image_creation_no_user'
image_file = FileStorage(
stream=open('fast_style_transfer/source-images/cape-flattery.jpg'))
image = Image.create(image_file)
self.assertEqual(image.image_id, 9)
self.assertIsInstance(image, Image)
self.assertTrue(image.is_public)
self.assertEqual(image.file_extension, 'jpg')
self.assertIsNone(image.user_id)
print '+ passed'
def save_data_to_db(data):
Image.create(date=data['date'], filename=data['filename'])