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 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
|
require 'test/unit'
require 'numru/netcdf'
include NumRu
class TestFactorOffset < Test::Unit::TestCase
def setup
@s = 'tmp.nc'
f = NetCDF.create(@s)
nx = 10
d = f.def_dim('x',nx)
v1 = f.def_var('v1','sint',[d])
v2 = f.def_var('v2','sint',[d])
v3 = f.def_var('v3','int',[d])
v1.put_att('scale_factor',0.1,'sfloat')
v1.put_att('add_offset',100.0,'sfloat')
v2.put_att('scale_factor',0.1,'sfloat')
v2.put_att('add_offset',100.0,'sfloat')
v3.put_att('scale_factor',0.1,'sfloat')
v3.put_att('add_offset',100.0,'sfloat')
f.enddef
v1.put( NArray.sint(nx).indgen!+100 )
v2.scaled_put( NArray.float(nx).indgen!+100 )
v3.scaled_put( NArray.float(nx).indgen!+100 )
f.close
end
def teardown
if File.exist?(@s)
begin
File.delete(@s)
rescue
p $!
end
end
end
def test_originally_uscaled
f = NetCDF.open(@s)
v1 = f.var('v1')
v11 = v1.get
assert_equal v11,
NArray[ 100, 101, 102, 103, 104, 105, 106, 107, 108, 109 ].to_i
v12 = v1.scaled_get
assert_equal v12,
NArray[ 110.0, 110.1, 110.2, 110.3, 110.4, 110.5, 110.6, 110.7, 110.8, 110.9 ].to_type(NArray::SFLOAT)
f.close
end
def test_originally_scaled
f = NetCDF.open(@s)
v2 = f.var('v2')
v21 = v2.get
assert_equal v21,
NArray[ 0, 10, 20, 30, 40, 50, 60, 70, 80, 90 ].to_i
v22 = v2.scaled_get
assert_equal v22,
NArray[ 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0 ].to_type(NArray::SFLOAT)
f.close
end
def test_originally_scaled_int_to_double
f = NetCDF.open(@s)
v3 = f.var('v3')
v31 = v3.get
assert_equal v31,
NArray[ 0, 10, 20, 30, 40, 50, 60, 70, 80, 90 ].to_i
v32 = v3.scaled_get
assert_equal v32,
NArray[ 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0 ].to_type(NArray::SFLOAT)
f.close
end
def test_unpack_type_fixed_to_sfloat
f = NetCDF.open(@s)
v3 = f.var('v3')
NetCDFVar.unpack_type = NArray::SFLOAT
v33 = v3.scaled_get
assert_equal v33,
NArray[ 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0 ].to_type(NArray::SFLOAT)
NetCDFVar.unpack_type = NArray::INT
v33 = v3.scaled_get
assert_equal v33,
NArray[ 100, 101, 102, 103, 104, 105, 106, 107, 108, 109 ].to_i
end
end
|