Added formatting for Ruby set objects (patch by Richard Hall)

lib/awesome_print/formatter.rb

@@ -9,7 +9,7 @@ require "shellwords"
 module AwesomePrint
   class Formatter
 
-    CORE = [ :array, :hash, :class, :file, :dir, :bigdecimal, :rational, :struct, :method, :unboundmethod ]
+    CORE = [ :array, :bigdecimal, :class, :dir, :file, :hash, :method, :rational, :set, :struct, :unboundmethod ]
     DEFAULT_LIMIT_SIZE = 7
 
     def initialize(inspector)
@@ -162,6 +162,12 @@ module AwesomePrint
       end
     end
 
+    # Format a set.
+    #------------------------------------------------------------------------------
+    def awesome_set(s)
+      awesome_array(s.to_a)
+    end
+
     # Format a Struct.
     #------------------------------------------------------------------------------
     def awesome_struct(s)

spec/formats_spec.rb

@@ -1,6 +1,7 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 require "bigdecimal"
 require "rational"
+require "set"
 
 describe "AwesomePrint" do
   before do
@@ -511,6 +512,34 @@ EOS
     end
   end
 
+  #------------------------------------------------------------------------------
+  describe "Set" do
+    before do
+      @arr = [1, :two, "three" ]
+      @set = Set.new(@arr)
+    end
+
+    it "empty set" do
+      Set.new.ai.should == [].ai
+    end
+
+    it "plain multiline" do
+      @set.ai(:plain => true).should == @arr.ai(:plain => true)
+    end
+
+    it "plain multiline indented" do
+      @set.ai(:plain => true, :indent => 1).should == @arr.ai(:plain => true, :indent => 1)
+    end
+
+    it "plain single line" do
+      @set.ai(:plain => true, :multiline => false).should == @arr.ai(:plain => true, :multiline => false)
+    end
+
+    it "colored multiline (default)" do
+      @set.ai.should == @arr.ai
+    end
+  end
+
   #------------------------------------------------------------------------------
   describe "Struct" do
     before do