binary_decision_tree 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cd430f35feb7cf78297db1208f4678ad0cb192f3
-  data.tar.gz: 4777835269c7eca78d13d3694e11070c2b8f5729
+  metadata.gz: 788140290951a2efc6b9ea9d01f74a473d2d2472
+  data.tar.gz: ff066dc9ad48d7dea86df354b7465561030c43ce
 SHA512:
-  metadata.gz: 7998bc2ad37b1d2b0652ac4b4117e199679a64b2594cafa3a2594e2c3b92c71e8b4c2b1a8b330f60c21d60b40d75cf546517d27d3f9c89aacd2bc370773fd81e
-  data.tar.gz: 000651f02ad554ab4ab1350936e16bba13e54f47cb8a417301c70bd9fbfcfd209f2fbf6a326c97db9162e2e9cc4fb6bb775c80607fe77377d281ac2d0bb582db
+  metadata.gz: 215dff89705f8acbb8d9be7dd0e1de1423e03bb2898e8337829775bbfc24bf51c3ef50019470b85524c026f851b50242585fd624337ddb8798c825cee74382dd
+  data.tar.gz: 06d2a3284a398b1c119a2e665aa3962eedbb9e114025fc932deb3b687df500ee7f08e399d0c342dc9237fd77d499feee09ea9a6ebd07a7fd71de711d036b00e8

data/lib/binary_decision_tree/marshalled_tree.rb

@@ -27,8 +27,8 @@ module BinaryDecisionTree
       new(depth, decisions, mask)
     end
 
-    def to_tree
-      tree = Tree.new(depth)
+    def to_tree(tree_class: Tree)
+      tree = tree_class.new(depth)
 
       (2**tree.depth).times do |i|
         next if i == 0

data/lib/binary_decision_tree/tree.rb

@@ -2,9 +2,9 @@ module BinaryDecisionTree
   class Tree
     attr_reader :depth
 
-    def initialize(depth)
+    def initialize(depth, node_class: Node)
       @depth = depth
-      @nodes = Array.new(size) {|i| i == 0 ? nil : Node.new(self, i)}
+      @nodes = Array.new(size) {|i| i == 0 ? nil : node_class.new(self, i)}
     end
 
     def root

data/lib/binary_decision_tree/version.rb

@@ -1,3 +1,3 @@
 module BinaryDecisionTree
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/test/unit/binary_decision_tree/marshalled_tree_test.rb

@@ -1,6 +1,8 @@
 require_relative '../../../test/test_helper'
 
 module BinaryDecisionTree
+  class SubTree < Tree; end
+
   class MarshalledTreeTest < MiniTest::Spec
     before do
       @tree = Tree.new(6)
@@ -8,13 +10,11 @@ module BinaryDecisionTree
         next if i == 0
         @tree.at(i).decision = rand(2)
       end
+
+      @marshalled_tree = MarshalledTree.from_tree(@tree)
     end
 
     describe "marshalling a tree" do
-      before do
-        @marshalled_tree = MarshalledTree.from_tree(@tree)
-      end
-
       it "should marshal and unmarshal to the same tree" do
         assert_same_tree @tree, @marshalled_tree.to_tree
       end
@@ -24,5 +24,12 @@ module BinaryDecisionTree
         assert_same_tree @tree, another_tree
       end
     end
+
+    describe "using a different tree class" do
+      it "should instantiate the correct object type" do
+        assert_equal Tree, @marshalled_tree.to_tree.class
+        assert_equal SubTree, @marshalled_tree.to_tree(tree_class: SubTree).class
+      end
+    end
   end
 end

data/test/unit/binary_decision_tree/tree_test.rb

@@ -0,0 +1,14 @@
+require_relative '../../../test/test_helper'
+
+module BinaryDecisionTree
+  class SubNode < Node; end
+
+  class TreeTest < MiniTest::Spec
+    describe "creating a new tree" do
+      it "should allow an optional node class" do
+        assert_equal Node, Tree.new(1).root.class
+        assert_equal SubNode, Tree.new(1, node_class: SubNode).root.class
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: binary_decision_tree
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Jason Haruska
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-04-30 00:00:00.000000000 Z
+date: 2014-05-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -104,6 +104,7 @@ files:
 - test/test_helper.rb
 - test/unit/binary_decision_tree/marshalled_tree_test.rb
 - test/unit/binary_decision_tree/node_test.rb
+- test/unit/binary_decision_tree/tree_test.rb
 homepage: http://github.com/haruska/binary_decision_tree
 licenses:
 - MIT
@@ -132,3 +133,4 @@ test_files:
 - test/test_helper.rb
 - test/unit/binary_decision_tree/marshalled_tree_test.rb
 - test/unit/binary_decision_tree/node_test.rb
+- test/unit/binary_decision_tree/tree_test.rb