Fix tests for pygments.rb 0.3.x

2 files changed, 5 insertions, 7 deletions

diff --git a/test/test_blob.rb b/test/test_blob.rb
index 333b0ce..46b1e0f 100644
--- a/test/test_blob.rb
+++ b/test/test_blob.rb
@@ -271,7 +271,7 @@ class TestBlob < Test::Unit::TestCase
     assert !blob("Binary/github.po").indexable?
     assert !blob("Binary/linguist.gem").indexable?
 
-    # large binary blobs should fail on size check first, not call 
+    # large binary blobs should fail on size check first, not call
     # into charlock_holmes and alloc big buffers for testing encoding
     b = blob("Binary/octocat.ai")
     b.expects(:binary?).never
@@ -291,11 +291,10 @@ class TestBlob < Test::Unit::TestCase
   end
 
   def test_colorize
-    assert_equal <<-HTML, blob("Ruby/foo.rb").colorize
+    assert_equal <<-HTML.chomp, blob("Ruby/foo.rb").colorize
 <div class="highlight"><pre><span class="k">module</span> <span class="nn">Foo</span>
 <span class="k">end</span>
-</pre>
-</div>
+</pre></div>
     HTML
   end
 
diff --git a/test/test_language.rb b/test/test_language.rb
index a6c08a4..ae0071d 100644
--- a/test/test_language.rb
+++ b/test/test_language.rb
@@ -319,12 +319,11 @@ class TestLanguage < Test::Unit::TestCase
 
 
   def test_colorize
-    assert_equal <<-HTML, Language['Ruby'].colorize("def foo\n  'foo'\nend\n")
+    assert_equal <<-HTML.chomp, Language['Ruby'].colorize("def foo\n  'foo'\nend\n")
 <div class="highlight"><pre><span class="k">def</span> <span class="nf">foo</span>
   <span class="s1">&#39;foo&#39;</span>
 <span class="k">end</span>
-</pre>
-</div>
+</pre></div>
     HTML
   end
 end