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
|
# Test of opening and closing and opening
import logging
import os.path
import shutil
import sys
import tempfile
import unittest
import rasterio
logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
log = logging.getLogger('rasterio.tests')
class RevolvingDoorTest(unittest.TestCase):
def setUp(self):
self.tempdir = tempfile.mkdtemp()
def tearDown(self):
shutil.rmtree(self.tempdir)
def test_write_colormap_revolving_door(self):
with rasterio.open('tests/data/shade.tif') as src:
shade = src.read(1)
meta = src.meta
tiffname = os.path.join(self.tempdir, 'foo.tif')
with rasterio.open(tiffname, 'w', **meta) as dst:
dst.write(shade, indexes=1)
with rasterio.open(tiffname) as src:
pass
|
