def test_raise_on_invalid_key(self):
# given
accumulator = CUBADataAccumulator()
accumulator.append(DataContainer(VELOCITY=(0.1, 0.2, 0.3)))
# when then
with self.assertRaises(KeyError):
accumulator[CUBA.MASS]
def test_data_conainter_with_tuple_values(self):
# given
accumulator = CUBADataAccumulator()
# when
accumulator.append(DataContainer(VELOCITY=(0.1, 0.2, 0.3)))
# then
self.assertEqual(len(accumulator), 1)
self.assertEqual(accumulator.keys, set([CUBA.VELOCITY]))
assert_array_equal(
vtk_to_numpy(accumulator[CUBA.VELOCITY]),
[(0.1, 0.2, 0.3)])
def test_accumulate_with_missing_values(self):
# given
accumulator = CUBADataAccumulator()
# when
accumulator.append(DataContainer())
# then
self.assertEqual(len(accumulator), 1)
# when
accumulator.append(
create_data_container(
restrict=[CUBA.NAME, CUBA.TEMPERATURE], constant=7))
# then
self.assertEqual(len(accumulator), 2)
self.assertEqual(accumulator.keys, set([CUBA.TEMPERATURE]))
array = vtk_to_numpy(accumulator[CUBA.TEMPERATURE])
assert_array_equal(
array, [numpy.nan, dummy_cuba_value(CUBA.TEMPERATURE, constant=7)])
def test_accumulate_on_keys(self):
# given
cuds_data = [create_data_container(constant=i) for i in range(10)]
# when
accumulator = CUBADataAccumulator(keys=[CUBA.NAME, CUBA.TEMPERATURE])
# then
self.assertEqual(len(accumulator), 0)
self.assertEqual(accumulator.keys, set([CUBA.TEMPERATURE]))
# when
for data in cuds_data:
accumulator.append(data)
# then
self.assertEqual(len(accumulator), 10)
self.assertEqual(accumulator.keys, set([CUBA.TEMPERATURE]))
assert_array_equal(
vtk_to_numpy(accumulator[CUBA.TEMPERATURE]),
[dummy_cuba_value(
CUBA.TEMPERATURE, constant=i) for i in range(10)])
def test_accumulate_and_expand(self):
# given
accumulator = CUBADataAccumulator()
# when
accumulator.append(create_data_container(restrict=[CUBA.MASS]))
# then
self.assertEqual(len(accumulator), 1)
# when
accumulator.append(
create_data_container(restrict=[CUBA.NAME, CUBA.TEMPERATURE]))
# then
self.assertEqual(len(accumulator), 2)
self.assertEqual(accumulator.keys, set([CUBA.MASS, CUBA.TEMPERATURE]))
assert_array_equal(
vtk_to_numpy(accumulator[CUBA.TEMPERATURE]),
[numpy.nan, dummy_cuba_value(CUBA.TEMPERATURE)])
assert_array_equal(
vtk_to_numpy(accumulator[CUBA.MASS]),
[dummy_cuba_value(CUBA.MASS), numpy.nan])
def test_accumulate(self):
# given
cuds_data = [create_data_container(constant=i) for i in range(10)]
# when
accumulator = CUBADataAccumulator()
for data in cuds_data:
accumulator.append(data)
# then
self.assertEqual(len(accumulator), 10)
expected_cuba = set(CUBA) & supported_cuba()
self.assertEqual(accumulator.keys, expected_cuba)
for cuba in expected_cuba:
default = dummy_cuba_value(cuba)
if isinstance(default, numpy.ndarray):
new_shape = (10,) + default.shape
assert_array_equal(
vtk_to_numpy(accumulator[cuba]).reshape(new_shape),
[dummy_cuba_value(cuba, constant=i) for i in range(10)])
else:
assert_array_equal(
vtk_to_numpy(accumulator[cuba]),
[dummy_cuba_value(cuba, constant=i) for i in range(10)])
