importing from rubyforge & cleanup

test/test_decisiontree.rb

@@ -0,0 +1,26 @@
+#The MIT License
+
+###Copyright (c) 2007 Ilya Grigorik <ilya AT fortehost DOT com>
+
+require File.dirname(__FILE__) + '/test_helper.rb'
+require 'decisiontree'
+
+class TestDecisionTree < Test::Unit::TestCase
+
+  def setup
+    @labels = %w(sun rain)
+    @data = [
+        [1, 0, 1],
+        [0, 1, 0]
+    ]
+  end
+  
+  def test_truth
+    dec_tree = DecisionTree::ID3Tree.new(@labels, @data, 1, :discrete)
+    dec_tree.train
+    
+    assert 1, dec_tree.predict([1, 0])
+    assert 0, dec_tree.predict([0, 1])
+  end
+end
+

test/test_helper.rb

@@ -0,0 +1,2 @@
+require 'test/unit'
+require File.dirname(__FILE__) + '/../lib/decisiontree'