def test_wood_design_text(self): result = show_preview(1, 1, 1, "Hello") known_result = Image.open("wood_design_text.png") img = BytesIO() known_result.save(img, format='PNG') img = img.getvalue() known_result = base64.b64encode(img) self.assertEqual(result, known_result)
def show_preview(): """Loads the image by applying customizations Returns: preview_image: customized image """ model_id = request.args.get('model_id') wood_id = request.args.get('wood_id') design_id = request.args.get('design_id') message = request.args.get('message') preview_image = preview.show_preview(model_id, wood_id, design_id, message) return jsonify({"preview_image": preview_image.decode('utf-8')})