def test_setting_units(name, unit_str):
fields = ModelDataFields()
fields.new_field_location("node", 12)
fields.add_field(name, np.empty(12), at="node", units=unit_str[::-1])
assert fields.field_units("node", name) == unit_str[::-1]
fields["node"].set_units(name, unit_str)
assert fields.field_units("node", name) == unit_str
def test_nd_field():
"""Test creating fields that are nd in shape."""
fields = ModelDataFields()
fields.new_field_location('node', 12)
fields.add_field('new_value', np.ones((12,4,5)), at='node')
fields.add_field('newer_value', np.ones((12,4)), at='node')
assert_raises(ValueError, fields.add_field, 'newest_value', np.ones((13,4,5)), at='node')
assert_raises(ValueError, fields.add_field, 'newestest_value', np.ones((13)), at='node')
def test_nd_field():
"""Test creating fields that are nd in shape."""
fields = ModelDataFields()
fields.new_field_location('node', 12)
fields.add_field('new_value', np.ones((12, 4, 5)), at='node')
fields.add_field('newer_value', np.ones((12, 4)), at='node')
assert_raises(ValueError,
fields.add_field,
'newest_value',
np.ones((13, 4, 5)),
at='node')
assert_raises(ValueError,
fields.add_field,
'newestest_value',
np.ones((13)),
at='node')
def test_nd_field():
"""Test creating fields that are nd in shape."""
fields = ModelDataFields()
fields.new_field_location("node", 12)
fields.add_field("new_value", np.ones((12, 4, 5)), at="node")
fields.add_field("newer_value", np.ones((12, 4)), at="node")
with pytest.raises(ValueError):
fields.add_field("newest_value", np.ones((13, 4, 5)), at="node")
with pytest.raises(ValueError):
fields.add_field("newestest_value", np.ones((13)), at="node")
def test_nd_field():
"""Test creating fields that are nd in shape."""
fields = ModelDataFields()
fields.new_field_location("node", 12)
fields.add_field("new_value", np.ones((12, 4, 5)), at="node")
fields.add_field("newer_value", np.ones((12, 4)), at="node")
with pytest.raises(ValueError):
fields.add_field("newest_value", np.ones((13, 4, 5)), at="node")
with pytest.raises(ValueError):
fields.add_field("newestest_value", np.ones((13)), at="node")
