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")
