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
|
import unittest
import netCDF4
from numpy.testing import assert_array_equal
import numpy as np
import pathlib
test_dir = pathlib.Path(__file__).parent
class Test_Unsigned(unittest.TestCase):
"""
Test autoconversion to unsigned ints when _Unsigned attribute is True.
This attribute is is set by netcdf-java to designate unsigned
integer data stored with a signed integer type in netcdf-3.
If _Unsigned=True, a view to the data as unsigned integers is returned.
set_autoscale can be used to turn this off (default is on)
See issue #656 (pull request #658).
"""
def test_unsigned(self):
with netCDF4.Dataset(test_dir / "ubyte.nc") as f:
data = f['ub'][:]
assert data.dtype.str[1:] == 'u1'
assert_array_equal(data,np.array([0,255],np.uint8))
f.set_auto_scale(False)
data2 = f['ub'][:]
assert data2.dtype.str[1:] == 'i1'
assert_array_equal(data2,np.array([0,-1],np.int8))
data = f['sb'][:]
assert data.dtype.str[1:] == 'i1'
# issue #1232 _Unsigned='false' is same as not having _Unsigned set.
data = f['sb2'][:]
assert data.dtype.str[1:] == 'i1'
# issue 671
with netCDF4.Dataset(test_dir / "issue671.nc") as f:
data1 = f['soil_moisture'][:]
assert np.ma.isMA(data1)
f.set_auto_scale(False)
data2 = f['soil_moisture'][:]
assert data1.mask.sum() == data2.mask.sum()
# issue 794
# test that valid_min/valid_max/_FillValue are
# treated as unsigned integers.
with netCDF4.Dataset(test_dir / "20171025_2056.Cloud_Top_Height.nc") as f:
data = f['HT'][:]
assert data.mask.sum() == 57432
assert int(data.max()) == 15430
assert int(data.min()) == 0
assert data.dtype == np.float32
if __name__ == '__main__':
unittest.main()
|
