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