GDAL Trunk CI Failure

gdal-trunk-ci-failure.py

import affine
import rasterio
from rasterio import shutil as rio_shutil
from rasterio.enums import ColorInterp


path_4band_no_colorinterp = '4band-no-ci.tif'


# This seems to be a GDAL default.  GDAL 2.2 allows 'undefined' for first
# band, but older versions silently convert to 'gray'
undefined_ci = (
    ColorInterp.gray,
    ColorInterp.undefined,
    ColorInterp.undefined,
    ColorInterp.undefined)


profile = {
    'height': 10,
    'width': 10,
    'count': 4,
    'dtype': rasterio.ubyte,
    'transform': affine.Affine(1, 0.0, 0,
                               0.0, -1, 1),
    'driver': 'GTiff',
    'photometric': 'minisblack'
}

with rasterio.open(path_4band_no_colorinterp, 'w', **profile) as src:
    src.colorinterp = undefined_ci

# Ensure color interp was set.
with rasterio.open(path_4band_no_colorinterp) as src:
    if src.colorinterp != undefined_ci:
        raise ValueError(
            "Didn't properly set color interpretation.  GDAL can "
            "forcefully make assumptions.")


for ci in ColorInterp.__members__.values():

    with rasterio.open(path_4band_no_colorinterp, 'r+') as src:
        all_ci = list(src.colorinterp)
        # Adjust CI for first band.  This appears to be failing on GDAL
        # trunk for 'ColorInterp.gray' and 'ColorInterp.grey', which are
        # aliases for the same value.
        all_ci[1] = ci
        src.colorinterp = all_ci

    with rasterio.open(path_4band_no_colorinterp) as src:
        assert src.colorinterp[1] == ci