adding yields for package workbook, workbook styles and cols collection on worksheet to make charting easier.

1 files changed, 12 insertions, 1 deletions

diff --git a/test/workbook/worksheet/tc_worksheet.rb b/test/workbook/worksheet/tc_worksheet.rb
index 6bdf5f20..b6865bd6 100644
--- a/test/workbook/worksheet/tc_worksheet.rb
+++ b/test/workbook/worksheet/tc_worksheet.rb
@@ -51,14 +51,25 @@ class TestWorksheet < Test::Unit::TestCase
     @ws.add_row [1,2,3,4]
     @ws.add_row [1,2,3,4]
     @ws.add_row [1,2,3,4]
-    @ws.col_style 1, 1, :row_offset=>1
+    @ws.col_style (1..2), 1, :row_offset=>1
     @ws.rows[(1..-1)].each do | r | 
       assert_equal(r.cells[1].style, 1)
+      assert_equal(r.cells[2].style, 1)
     end
     assert_equal(@ws.rows.first.cells[1].style, 0)
     assert_equal(@ws.rows.first.cells[0].style, 0)
   end
 
+  def test_cols
+    @ws.add_row [1,2,3,4]
+    @ws.add_row [1,2,3,4]
+    @ws.add_row [1,2,3,4]
+    @ws.add_row [1,2,3,4]
+    c = @ws.cols[1]
+    assert_equal(c.size, 4)    
+    assert_equal(c[0].value, 2)
+  end
+
   def test_row_style
     @ws.add_row [1,2,3,4]
     @ws.add_row [1,2,3,4]