Fix 90, 180 and 270 degrees rotation transformations.
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
@@ -6,8 +6,9 @@ from documents.tests import GenericDocumentTestCase
|
||||
|
||||
from ..models import Transformation
|
||||
from ..transformations import (
|
||||
BaseTransformation, TransformationCrop, TransformationResize,
|
||||
TransformationRotate, TransformationZoom
|
||||
BaseTransformation, TransformationCrop, TransformationLineArt,
|
||||
TransformationResize, TransformationRotate, TransformationRotate90,
|
||||
TransformationRotate180, TransformationRotate270, TransformationZoom
|
||||
)
|
||||
|
||||
from .literals import (
|
||||
@@ -154,3 +155,37 @@ class TransformationTestCase(GenericDocumentTestCase):
|
||||
)
|
||||
|
||||
self.assertTrue(document_page.generate_image().startswith('page'))
|
||||
|
||||
def test_lineart_transformations(self):
|
||||
document_page = self.document.pages.first()
|
||||
|
||||
Transformation.objects.add_for_model(
|
||||
obj=document_page, transformation=TransformationLineArt,
|
||||
arguments={}
|
||||
)
|
||||
|
||||
self.assertTrue(document_page.generate_image().startswith('page'))
|
||||
|
||||
def test_rotate_transformations(self):
|
||||
document_page = self.document.pages.first()
|
||||
|
||||
Transformation.objects.add_for_model(
|
||||
obj=document_page, transformation=TransformationRotate90,
|
||||
arguments={}
|
||||
)
|
||||
|
||||
self.assertTrue(document_page.generate_image().startswith('page'))
|
||||
|
||||
Transformation.objects.add_for_model(
|
||||
obj=document_page, transformation=TransformationRotate180,
|
||||
arguments={}
|
||||
)
|
||||
|
||||
self.assertTrue(document_page.generate_image().startswith('page'))
|
||||
|
||||
Transformation.objects.add_for_model(
|
||||
obj=document_page, transformation=TransformationRotate270,
|
||||
arguments={}
|
||||
)
|
||||
|
||||
self.assertTrue(document_page.generate_image().startswith('page'))
|
||||
|
||||
@@ -210,8 +210,9 @@ class TransformationRotate(BaseTransformation):
|
||||
if self.degrees == 0:
|
||||
return self.image
|
||||
|
||||
if self.fillcolor:
|
||||
fillcolor = ImageColor.getrgb(self.fillcolor)
|
||||
fillcolor_value = getattr(self, 'fillcolor', None)
|
||||
if fillcolor_value:
|
||||
fillcolor = ImageColor.getrgb(fillcolor_value)
|
||||
else:
|
||||
fillcolor = None
|
||||
|
||||
|
||||
Reference in New Issue
Block a user