Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# -*- coding: utf-8 -*-
"""
***************************************************************************
---------------------
Date : 30.11.2017
Copyright : (C) 2017 by Benjamin Jakimow
Email : benjamin jakimow at geo dot hu-berlin dot de
***************************************************************************
* *
* 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. *
* *
***************************************************************************
"""
# noinspection PyPep8Naming
import os, sys, unittest, configparser
from timeseriesviewer.tests import initQgisApplication
from PyQt5.QtGui import *
from PyQt5.QtCore import *
import unittest, tempfile
class TestInMemoryData(unittest.TestCase):
"""
Tests for the GDAL/OGR VSI in-memory data
"""
def test_vsimem_raster(self):
QGIS_APP = initQgisApplication()
from osgeo import gdal
from timeseriesviewer.tests import TestObjects
from qgis.core import QgsCoordinateReferenceSystem, QgsRasterLayer
TestObjects.inMemoryImage()
# create an in-memory raster
driver = gdal.GetDriverByName('GTiff')
self.assertIsInstance(driver, gdal.Driver)
path = '/vsimem/inmemoryraster.tif'
dataSet = driver.Create(path, 100, 50, bands=3, eType=gdal.GDT_Int16)
self.assertIsInstance(dataSet, gdal.Dataset)
c = QgsCoordinateReferenceSystem('EPSG:32632')
dataSet.SetProjection(c.toWkt())
dataSet.SetGeoTransform([0, 1.0, 0, \
0, 0, -1.0])
dataSet.FlushCache()
dataSet = None
ds2 = gdal.Open(path)
self.assertIsInstance(ds2, gdal.Dataset)
layer = QgsRasterLayer(path)
self.assertIsInstance(layer, QgsRasterLayer)
self.assertTrue(layer.isValid())
def test_vsimem_raster2(self):
from osgeo import gdal
from qgis.core import QgsCoordinateReferenceSystem, QgsRasterLayer
# create an in-memory raster
driver = gdal.GetDriverByName('GTiff')
assert isinstance(driver, gdal.Driver)
path = '/vsimem/inmemoryraster.tif'
dataSet = driver.Create(path, 100, 50, bands=3, eType=gdal.GDT_Int16)
assert isinstance(dataSet, gdal.Dataset)
c = QgsCoordinateReferenceSystem('EPSG:32632')
dataSet.SetProjection(c.toWkt())
dataSet.SetGeoTransform([0, 1.0, 0, 0, 0, -1.0])
dataSet.FlushCache()
dataSet = None
ds2 = gdal.Open(path)
assert isinstance(ds2, gdal.Dataset)
layer = QgsRasterLayer(path)
assert isinstance(layer, QgsRasterLayer)
assert layer.isValid()
if __name__ == '__main__':
unittest.main()