rasterio/tests/test_driver_management.py
2018-10-23 16:41:52 -06:00

52 lines
1.4 KiB
Python

import logging
import rasterio
from rasterio._env import driver_count
def test_drivers():
with rasterio.Env() as m:
assert driver_count() > 0
assert type(m) == rasterio.Env
assert driver_count() > 0
def test_drivers_bwd_compat():
with rasterio.Env() as m:
assert driver_count() > 0
assert type(m) == rasterio.Env
assert driver_count() > 0
def test_cpl_debug_true(tmpdir):
"""Setting CPL_DEBUG=True results in GDAL debug messages."""
log = logging.getLogger('rasterio._env')
log.setLevel(logging.DEBUG)
logfile = str(tmpdir.join('test.log'))
fh = logging.FileHandler(logfile)
log.addHandler(fh)
with rasterio.Env(CPL_DEBUG=True):
with rasterio.open("tests/data/RGB.byte.tif"):
pass
log = open(logfile).read()
assert "GDAL: GDALOpen(tests/data/RGB.byte.tif" in log
def test_cpl_debug_false(tmpdir):
"""Setting CPL_DEBUG=False results in no GDAL debug messages."""
log = logging.getLogger('rasterio._gdal')
log.setLevel(logging.DEBUG)
logfile = str(tmpdir.join('test.log'))
fh = logging.FileHandler(logfile)
log.addHandler(fh)
with rasterio.Env(CPL_DEBUG=False):
with rasterio.open("tests/data/RGB.byte.tif"):
pass
# Expect no debug messages from GDAL.
log = open(logfile).read()
assert "GDAL: GDALOpen(tests/data/RGB.byte.tif" not in log