rasterio/tests/test_dtypes.py
Sean Gillies 4e53e3ca2d Refactor to avoid unneeded numpy imports.
Closes #140.

Fairly easy because Numpy methods accept either strings like
`'uint8'` or types like `np.uint8` as dtypes. So do Rasterio
methods now.
2014-09-05 14:11:36 -06:00

15 lines
412 B
Python

import numpy as np
import rasterio.dtypes
def test_np_dt_uint8():
assert rasterio.dtypes.check_dtype(np.uint8)
def test_dt_ubyte():
assert rasterio.dtypes.check_dtype(rasterio.ubyte)
def test_gdal_name():
assert rasterio.dtypes._gdal_typename(rasterio.ubyte) == 'Byte'
assert rasterio.dtypes._gdal_typename(np.uint8) == 'Byte'
assert rasterio.dtypes._gdal_typename(np.uint16) == 'UInt16'