test/neural_network/backpropagation_test.rb in ai4r-1.3 vs test/neural_network/backpropagation_test.rb in ai4r-1.4

@@ -26,9 +26,32 @@
   module NeuralNetwork
 
 
     class BackpropagationTest < Test::Unit::TestCase
 
+      
+      def test_init_network
+        net_4_2 = Backpropagation.new([4, 2]).init_network
+        assert_equal [[1.0, 1.0, 1.0, 1.0, 1.0], [1.0, 1.0]], 
+          net_4_2.activation_nodes
+        assert_equal 1, net_4_2.weights.size
+        assert_equal 5, net_4_2.weights.first.size
+        net_4_2.weights.first.each do |weights_n|
+          assert_equal 2, weights_n.size  
+        end
+        
+        net_2_2_1 = Backpropagation.new([2, 2, 1]).init_network
+        assert_equal [[1.0, 1.0, 1.0], [1.0, 1.0, 1.0], [1.0]], 
+          net_2_2_1.activation_nodes
+        assert_equal 2, net_2_2_1.weights.size
+        assert_equal 3, net_2_2_1.weights.first.size
+        
+        net_2_2_1.disable_bias = true
+        net_2_2_1_no_bias = net_2_2_1.init_network
+        assert_equal [[1.0, 1.0], [1.0, 1.0], [1.0]], 
+          net_2_2_1_no_bias.activation_nodes
+      end
+      
       def test_eval
         #Test set 1
         net = Backpropagation.new([3, 2])
         y = net.eval([3, 2, 3])
         assert y.length == 2
\ No newline at end of file