class Seahorse::Model::Shapes::Shape

def shape_at(key)
  if @definition[key]
    shape_for(@definition[key])
  else
    raise ArgumentError, "expected shape definition at #{key.inspect}"
  end
end