more tests

test/functional_test.rb

@@ -2,13 +2,28 @@ require 'test/unit'
 require 'functional'
 
 class FunTest < Test::Unit::TestCase
+	M = 0..100
+
+	def doit_fun m, &exe
+		f = m.to_fun
+		yield f
+		f.to_a
+	end
+
 	def test_to_fun_exists
 		assert_respond_to Object, :to_fun
 	end
 
-	def test_to_fun_to_a
-		assert_equal (0..100).to_a, (0..100).to_fun.to_a
+	def test_to_a
+		assert_equal M.to_a, doit_fun( M) {|x| x }
 	end
 
-	
+	def test_collect
+		l = lambda {|x| x*2}
+		assert_equal M.collect( &l), doit_fun( M) {|x| x.collect( &l) }
+	end
+
+	def test_inject
+		assert_equal M.inject( 0) {|i,j| i+j }, M.to_fun.inject( 0) {|i,j| i+j }
+	end
 end