TokensProxy: wrap up

4 files changed, 4 insertions, 4 deletions

diff --git a/test/functional/basic.rb b/test/functional/basic.rb
index 2654359..5d03513 100755
--- a/test/functional/basic.rb
+++ b/test/functional/basic.rb
@@ -26,7 +26,7 @@ class BasicTest < Test::Unit::TestCase
   ].flatten
   def test_simple_scan
     assert_nothing_raised do
-      assert_equal RUBY_TEST_TOKENS, CodeRay.scan(RUBY_TEST_CODE, :ruby).to_ary
+      assert_equal RUBY_TEST_TOKENS, CodeRay.scan(RUBY_TEST_CODE, :ruby).tokens
     end
   end
   
diff --git a/test/unit/count.rb b/test/unit/count.rb
index ad61291..448e8f1 100644
--- a/test/unit/count.rb
+++ b/test/unit/count.rb
@@ -9,7 +9,7 @@ class CountTest < Test::Unit::TestCase
 # a minimal Ruby program
 puts "Hello world!"
     RUBY
-    assert_equal 11, tokens.encode_with(:count)
+    assert_equal 11, tokens.encode(:count)
   end
   
 end
\ No newline at end of file
diff --git a/test/unit/null.rb b/test/unit/null.rb
index ea516d8..d3a9b0d 100644
--- a/test/unit/null.rb
+++ b/test/unit/null.rb
@@ -8,7 +8,7 @@ class NullTest < Test::Unit::TestCase
 puts "Hello world!"
     RUBY
     tokens = CodeRay.scan ruby, :ruby
-    assert_equal '', tokens.encode_with(:null)
+    assert_equal '', tokens.encode(:null)
   end
   
 end
\ No newline at end of file
diff --git a/test/unit/text.rb b/test/unit/text.rb
index 025881e..db086f5 100644
--- a/test/unit/text.rb
+++ b/test/unit/text.rb
@@ -8,7 +8,7 @@ class TextTest < Test::Unit::TestCase
 puts "Hello world!"
     RUBY
     tokens = CodeRay.scan ruby, :ruby
-    assert_equal ruby, tokens.encode_with(:text)
+    assert_equal ruby, tokens.encode(:text)
   end
   
 end
\ No newline at end of file