class Tests(tests.IrisTest):
def setUp(self):
self.values = np.array((10., 12., 16., 9.))
self.measure = CellMeasure(self.values, units='m^2',
standard_name='cell_area',
long_name='measured_area',
var_name='area',
attributes={'notes': '1m accuracy'},
measure='area')
def test_invalid_measure(self):
msg = "measure must be 'area' or 'volume', not length"
with self.assertRaisesRegexp(ValueError, msg):
self.measure.measure = 'length'
def test_set_measure(self):
v = 'volume'
self.measure.measure = v
self.assertEqual(self.measure.measure, v)
def test_data(self):
self.assertArrayEqual(self.measure.data, self.values)
def test_set_data(self):
new_vals = np.array((1., 2., 3., 4.))
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__int(self):
new_vals = np.array((1, 2, 3, 4), dtype=np.int32)
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__uint(self):
new_vals = np.array((1, 2, 3, 4), dtype=np.uint32)
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__lazy(self):
new_vals = as_lazy_data(np.array((1., 2., 3., 4.)))
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__int__lazy(self):
new_vals = as_lazy_data(np.array((1, 2, 3, 4), dtype=np.int32))
exp_emsg = "Cannot create cell measure with lazy data of type int32"
with self.assertRaisesRegexp(ValueError, exp_emsg):
self.measure.data = new_vals
def test_set_data__uint__lazy(self):
new_vals = as_lazy_data(np.array((1, 2, 3, 4), dtype=np.uint32))
exp_emsg = "Cannot create cell measure with lazy data of type uint32"
with self.assertRaisesRegexp(ValueError, exp_emsg):
self.measure.data = new_vals
def test_data_different_shape(self):
new_vals = np.array((1., 2., 3.))
msg = 'New data shape must match existing data shape.'
with self.assertRaisesRegexp(ValueError, msg):
self.measure.data = new_vals
def test_shape(self):
self.assertEqual(self.measure.shape, (4,))
def test_ndim(self):
self.assertEqual(self.measure.ndim, 1)
def test___getitem__(self):
sub_measure = self.measure[2]
self.assertArrayEqual(self.values[2], sub_measure.data)
def test___getitem__data_copy(self):
# Check that a sliced cell measure has independent data.
sub_measure = self.measure[1:3]
old_values = sub_measure.data.copy()
# Change the original one.
self.measure.data[:] = 0.0
# Check the new one has not changed.
self.assertArrayEqual(sub_measure.data, old_values)
def test_copy(self):
new_vals = np.array((7., 8.))
copy_measure = self.measure.copy(new_vals)
self.assertArrayEqual(copy_measure.data, new_vals)
def test_repr_other_metadata(self):
expected = (", long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'}")
self.assertEqual(self.measure._repr_other_metadata(), expected)
def test__str__(self):
expected = ("CellMeasure(array([10., 12., 16., 9.]), "
"measure=area, standard_name='cell_area', "
"units=Unit('m^2'), long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'})")
self.assertEqual(self.measure.__str__(), expected)
def test__repr__(self):
expected = ("CellMeasure(array([10., 12., 16., 9.]), "
"measure=area, standard_name='cell_area', "
"units=Unit('m^2'), long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'})")
self.assertEqual(self.measure.__repr__(), expected)
class Tests(tests.IrisTest):
def setUp(self):
self.values = np.array((10.0, 12.0, 16.0, 9.0))
self.measure = CellMeasure(
self.values,
units="m^2",
standard_name="cell_area",
long_name="measured_area",
var_name="area",
attributes={"notes": "1m accuracy"},
)
def test_invalid_measure(self):
msg = "measure must be 'area' or 'volume', got 'length'"
with self.assertRaisesRegex(ValueError, msg):
self.measure.measure = "length"
def test_set_measure(self):
v = "volume"
self.measure.measure = v
self.assertEqual(self.measure.measure, v)
def test_data(self):
self.assertArrayEqual(self.measure.data, self.values)
def test_set_data(self):
new_vals = np.array((1.0, 2.0, 3.0, 4.0))
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__int(self):
new_vals = np.array((1, 2, 3, 4), dtype=np.int32)
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__uint(self):
new_vals = np.array((1, 2, 3, 4), dtype=np.uint32)
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__lazy(self):
new_vals = as_lazy_data(np.array((1.0, 2.0, 3.0, 4.0)))
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_data_different_shape(self):
new_vals = np.array((1.0, 2.0, 3.0))
msg = "Require data with shape."
with self.assertRaisesRegex(ValueError, msg):
self.measure.data = new_vals
def test_shape(self):
self.assertEqual(self.measure.shape, (4,))
def test_ndim(self):
self.assertEqual(self.measure.ndim, 1)
def test___getitem__(self):
sub_measure = self.measure[2]
self.assertArrayEqual(self.values[2], sub_measure.data)
def test___getitem__data_copy(self):
# Check that a sliced cell measure has independent data.
sub_measure = self.measure[1:3]
old_values = sub_measure.data.copy()
# Change the original one.
self.measure.data[:] = 0.0
# Check the new one has not changed.
self.assertArrayEqual(sub_measure.data, old_values)
def test_copy(self):
new_vals = np.array((7.0, 8.0))
copy_measure = self.measure.copy(new_vals)
self.assertArrayEqual(copy_measure.data, new_vals)
def test___str__(self):
expected = "\n".join(
[
"CellMeasure : cell_area / (m^2)",
" data: [10., 12., 16., 9.]",
" shape: (4,)",
" dtype: float64",
" standard_name: 'cell_area'",
" long_name: 'measured_area'",
" var_name: 'area'",
" attributes:",
" notes '1m accuracy'",
" measure: 'area'",
]
)
self.assertEqual(self.measure.__str__(), expected)
def test___repr__(self):
expected = (
"<CellMeasure: cell_area / (m^2) "
"[10., 12., 16., 9.] shape(4,)>"
)
self.assertEqual(expected, self.measure.__repr__())
def test__eq__(self):
self.assertEqual(self.measure, self.measure)
class Tests(tests.IrisTest):
def setUp(self):
self.values = np.array((10., 12., 16., 9.))
self.measure = CellMeasure(self.values,
units='m^2',
standard_name='cell_area',
long_name='measured_area',
var_name='area',
attributes={'notes': '1m accuracy'},
measure='area')
def test_invalid_measure(self):
msg = "measure must be 'area' or 'volume', not length"
with self.assertRaisesRegexp(ValueError, msg):
self.measure.measure = 'length'
def test_set_measure(self):
v = 'volume'
self.measure.measure = v
self.assertEqual(self.measure.measure, v)
def test_data(self):
self.assertArrayEqual(self.measure.data, self.values)
def test_set_data(self):
new_vals = np.array((1., 2., 3., 4.))
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__int(self):
new_vals = np.array((1, 2, 3, 4), dtype=np.int32)
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__uint(self):
new_vals = np.array((1, 2, 3, 4), dtype=np.uint32)
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_set_data__lazy(self):
new_vals = as_lazy_data(np.array((1., 2., 3., 4.)))
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_data_different_shape(self):
new_vals = np.array((1., 2., 3.))
msg = 'Require data with shape.'
with self.assertRaisesRegexp(ValueError, msg):
self.measure.data = new_vals
def test_shape(self):
self.assertEqual(self.measure.shape, (4, ))
def test_ndim(self):
self.assertEqual(self.measure.ndim, 1)
def test___getitem__(self):
sub_measure = self.measure[2]
self.assertArrayEqual(self.values[2], sub_measure.data)
def test___getitem__data_copy(self):
# Check that a sliced cell measure has independent data.
sub_measure = self.measure[1:3]
old_values = sub_measure.data.copy()
# Change the original one.
self.measure.data[:] = 0.0
# Check the new one has not changed.
self.assertArrayEqual(sub_measure.data, old_values)
def test_copy(self):
new_vals = np.array((7., 8.))
copy_measure = self.measure.copy(new_vals)
self.assertArrayEqual(copy_measure.data, new_vals)
def test_repr_other_metadata(self):
expected = (", long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'}")
self.assertEqual(self.measure._repr_other_metadata(), expected)
def test__str__(self):
expected = ("CellMeasure(array([10., 12., 16., 9.]), "
"measure=area, standard_name='cell_area', "
"units=Unit('m^2'), long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'})")
self.assertEqual(self.measure.__str__(), expected)
def test__repr__(self):
expected = ("CellMeasure(array([10., 12., 16., 9.]), "
"measure=area, standard_name='cell_area', "
"units=Unit('m^2'), long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'})")
self.assertEqual(self.measure.__repr__(), expected)
def test__eq__(self):
self.assertEqual(self.measure, self.measure)
class Tests(tests.IrisTest):
def setUp(self):
self.values = np.array((10., 12., 16., 9.))
self.measure = CellMeasure(self.values, units='m^2',
standard_name='cell_area',
long_name='measured_area',
var_name='area',
attributes={'notes': '1m accuracy'},
measure='area')
def test_invalid_measure(self):
msg = "measure must be 'area' or 'volume', not length"
with self.assertRaisesRegexp(ValueError, msg):
self.measure.measure = 'length'
def test_set_measure(self):
v = 'volume'
self.measure.measure = v
self.assertEqual(self.measure.measure, v)
def test_data(self):
self.assertArrayEqual(self.measure.data, self.values)
def test_set_data(self):
new_vals = np.array((1., 2., 3., 4.))
self.measure.data = new_vals
self.assertArrayEqual(self.measure.data, new_vals)
def test_data_different_shape(self):
new_vals = np.array((1., 2., 3.))
msg = 'New data shape must match existing data shape.'
with self.assertRaisesRegexp(ValueError, msg):
self.measure.data = new_vals
def test_shape(self):
self.assertEqual(self.measure.shape, (4,))
def test_ndim(self):
self.assertEqual(self.measure.ndim, 1)
def test___getitem__(self):
sub_measure = self.measure[2]
self.assertArrayEqual(self.values[2], sub_measure.data)
def test_copy(self):
new_vals = np.array((7., 8.))
copy_measure = self.measure.copy(new_vals)
self.assertArrayEqual(copy_measure.data, new_vals)
def test_repr_other_metadata(self):
expected = (", long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'}")
self.assertEqual(self.measure._repr_other_metadata(), expected)
def test__str__(self):
expected = ("CellMeasure(array([ 10., 12., 16., 9.]), "
"measure=area, standard_name='cell_area', "
"units=Unit('m^2'), long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'})")
self.assertEqual(self.measure.__str__(), expected)
def test__repr__(self):
expected = ("CellMeasure(array([ 10., 12., 16., 9.]), "
"measure=area, standard_name='cell_area', "
"units=Unit('m^2'), long_name='measured_area', "
"var_name='area', attributes={'notes': '1m accuracy'})")
self.assertEqual(self.measure.__repr__(), expected)
