mirror of
https://github.com/qgis/QGIS.git
synced 2025-10-15 00:02:52 -04:00
149 lines
5.4 KiB
Python
149 lines
5.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""QGIS Unit tests for QgsComposerPicture.
|
|
|
|
.. note:: This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
"""
|
|
__author__ = '(C) 2015 by Nyall Dawson'
|
|
__date__ = '02/07/2015'
|
|
__copyright__ = 'Copyright 2015, The QGIS Project'
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
__revision__ = '$Format:%H$'
|
|
|
|
import qgis # NOQA
|
|
|
|
import os
|
|
import socketserver
|
|
import threading
|
|
import http.server
|
|
from qgis.PyQt.QtCore import QRectF
|
|
|
|
from qgis.core import (QgsComposerPicture,
|
|
QgsComposition,
|
|
QgsMapSettings,
|
|
QgsComposerMap,
|
|
QgsRectangle,
|
|
QgsCoordinateReferenceSystem,
|
|
QgsProject
|
|
)
|
|
from qgis.testing import start_app, unittest
|
|
from utilities import unitTestDataPath
|
|
from qgscompositionchecker import QgsCompositionChecker
|
|
|
|
start_app()
|
|
TEST_DATA_DIR = unitTestDataPath()
|
|
|
|
|
|
class TestQgsComposerPicture(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
# Bring up a simple HTTP server, for remote picture tests
|
|
os.chdir(unitTestDataPath() + '')
|
|
handler = http.server.SimpleHTTPRequestHandler
|
|
|
|
cls.httpd = socketserver.TCPServer(('localhost', 0), handler)
|
|
cls.port = cls.httpd.server_address[1]
|
|
|
|
cls.httpd_thread = threading.Thread(target=cls.httpd.serve_forever)
|
|
cls.httpd_thread.setDaemon(True)
|
|
cls.httpd_thread.start()
|
|
|
|
def __init__(self, methodName):
|
|
"""Run once on class initialization."""
|
|
unittest.TestCase.__init__(self, methodName)
|
|
|
|
TEST_DATA_DIR = unitTestDataPath()
|
|
self.pngImage = TEST_DATA_DIR + "/sample_image.png"
|
|
|
|
# create composition
|
|
self.composition = QgsComposition(QgsProject.instance())
|
|
self.composition.setPaperSize(297, 210)
|
|
|
|
self.composerPicture = QgsComposerPicture(self.composition)
|
|
self.composerPicture.setPicturePath(self.pngImage)
|
|
self.composerPicture.setSceneRect(QRectF(70, 70, 100, 100))
|
|
self.composerPicture.setFrameEnabled(True)
|
|
self.composition.addComposerPicture(self.composerPicture)
|
|
|
|
def testResizeZoom(self):
|
|
"""Test picture resize zoom mode."""
|
|
self.composerPicture.setResizeMode(QgsComposerPicture.Zoom)
|
|
|
|
checker = QgsCompositionChecker('composerpicture_resize_zoom', self.composition)
|
|
checker.setControlPathPrefix("composer_picture")
|
|
testResult, message = checker.testComposition()
|
|
|
|
assert testResult, message
|
|
|
|
@unittest.skip('test is broken for qt5/python3 - feature works')
|
|
def testRemoteImage(self):
|
|
"""Test fetching remote picture."""
|
|
self.composerPicture.setPicturePath('http://localhost:' + str(TestQgsComposerPicture.port) + '/qgis_local_server/logo.png')
|
|
|
|
checker = QgsCompositionChecker('composerpicture_remote', self.composition)
|
|
checker.setControlPathPrefix("composer_picture")
|
|
testResult, message = checker.testComposition()
|
|
|
|
self.composerPicture.setPicturePath(self.pngImage)
|
|
assert testResult, message
|
|
|
|
def testGridNorth(self):
|
|
"""Test syncing picture to grid north"""
|
|
|
|
composition = QgsComposition(QgsProject.instance())
|
|
|
|
composerMap = QgsComposerMap(composition)
|
|
composerMap.setNewExtent(QgsRectangle(0, -256, 256, 0))
|
|
composition.addComposerMap(composerMap)
|
|
|
|
composerPicture = QgsComposerPicture(composition)
|
|
composition.addComposerPicture(composerPicture)
|
|
|
|
composerPicture.setRotationMap(composerMap.id())
|
|
self.assertTrue(composerPicture.rotationMap() >= 0)
|
|
|
|
composerPicture.setNorthMode(QgsComposerPicture.GridNorth)
|
|
composerMap.setMapRotation(45)
|
|
self.assertEqual(composerPicture.pictureRotation(), 45)
|
|
|
|
# add an offset
|
|
composerPicture.setNorthOffset(-10)
|
|
self.assertEqual(composerPicture.pictureRotation(), 35)
|
|
|
|
def testTrueNorth(self):
|
|
"""Test syncing picture to true north"""
|
|
|
|
composition = QgsComposition(QgsProject.instance())
|
|
|
|
composerMap = QgsComposerMap(composition)
|
|
composerMap.setCrs(QgsCoordinateReferenceSystem.fromEpsgId(3575))
|
|
composerMap.setNewExtent(QgsRectangle(-2126029.962, -2200807.749, -119078.102, -757031.156))
|
|
composition.addComposerMap(composerMap)
|
|
|
|
composerPicture = QgsComposerPicture(composition)
|
|
composition.addComposerPicture(composerPicture)
|
|
|
|
composerPicture.setRotationMap(composerMap.id())
|
|
self.assertTrue(composerPicture.rotationMap() >= 0)
|
|
|
|
composerPicture.setNorthMode(QgsComposerPicture.TrueNorth)
|
|
self.assertAlmostEqual(composerPicture.pictureRotation(), 37.20, 1)
|
|
|
|
# shift map
|
|
composerMap.setNewExtent(QgsRectangle(2120672.293, -3056394.691, 2481640.226, -2796718.780))
|
|
self.assertAlmostEqual(composerPicture.pictureRotation(), -38.18, 1)
|
|
|
|
# rotate map
|
|
composerMap.setMapRotation(45)
|
|
self.assertAlmostEqual(composerPicture.pictureRotation(), -38.18 + 45, 1)
|
|
|
|
# add an offset
|
|
composerPicture.setNorthOffset(-10)
|
|
self.assertAlmostEqual(composerPicture.pictureRotation(), -38.18 + 35, 1)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|