Example #1
0
 def test_stoptrain_result1(self):
     """Test that stop_result is handled correctly."""
     stop_result = ({"test": 0}, 1)
     bi_sfa_node = SFABiNode(stop_result=stop_result, node_id="testing binode")
     assert bi_sfa_node.is_trainable()
     x = n.random.random((100, 10))
     train_result = bi_sfa_node.train(x)
     assert train_result == None
     assert bi_sfa_node.is_training()
     result = bi_sfa_node.stop_training()
     assert result == (None,) + stop_result
     assert bi_sfa_node.input_dim == 10
     assert bi_sfa_node.output_dim == 10
     assert bi_sfa_node.dtype == "float64"
Example #2
0
 def test_stoptrain_result1(self):
     """Test that stop_result is handled correctly."""
     stop_result = ({"test": 0}, 1)
     bi_sfa_node = SFABiNode(stop_result=stop_result,
                             node_id="testing binode")
     assert bi_sfa_node.is_trainable()
     x = n.random.random((100,10))
     train_result = bi_sfa_node.train(x)
     assert train_result is None
     assert bi_sfa_node.is_training()
     result = bi_sfa_node.stop_training()
     assert result == (None,) + stop_result
     assert bi_sfa_node.input_dim == 10
     assert bi_sfa_node.output_dim == 10
     assert bi_sfa_node.dtype == "float64"