class RecurrentConfigTest(unittest.TestCase):
"""
Tests RecurrentConfig
"""
def setUp(self):
self.net = NeuralNet()
self.net.init_layers(2, [1], 1)
self.rec_config = RecurrentConfig()
def test_apply_config(self):
self.assertRaises(
ValueError,
self.rec_config.apply_config, 'not neural net')
def test__apply_config(self):
print 'test__apply_config not yet implemented'
def test_fully_connect(self):
node = Node()
unode1 = Node()
unode2 = Node()
self.rec_config._fully_connect(node, [unode1, unode2])
conn = unode1.input_connections[0]
self.assertEqual(node, conn.lower_node)
self.assertEqual(unode1, conn.upper_node)
conn = unode2.input_connections[0]
self.assertEqual(node, conn.lower_node)
self.assertEqual(unode2, conn.upper_node)
def test_get_source_nodes(self):
self.assertEqual(True, isinstance(
self.rec_config.get_source_nodes(self.net),
NeuralNet))
def test_get_upper_nodes(self):
self.assertEqual(1, len(self.rec_config.get_upper_nodes(self.net)))
class RecurrentConfigTest(unittest.TestCase):
"""
Tests RecurrentConfig
"""
def setUp(self):
self.net = NeuralNet()
self.net.init_layers(2, [1], 1)
self.rec_config = RecurrentConfig()
def test_apply_config(self):
self.assertRaises(ValueError, self.rec_config.apply_config,
'not neural net')
def test__apply_config(self):
print('test__apply_config not yet implemented')
def test_fully_connect(self):
node = Node()
unode1 = Node()
unode2 = Node()
self.rec_config._fully_connect(node, [unode1, unode2])
conn = unode1.input_connections[0]
self.assertEqual(node, conn.lower_node)
self.assertEqual(unode1, conn.upper_node)
conn = unode2.input_connections[0]
self.assertEqual(node, conn.lower_node)
self.assertEqual(unode2, conn.upper_node)
def test_get_source_nodes(self):
self.assertEqual(
True,
isinstance(self.rec_config.get_source_nodes(self.net), NeuralNet))
def test_get_upper_nodes(self):
self.assertEqual(1, len(self.rec_config.get_upper_nodes(self.net)))
def setUp(self):
self.net = NeuralNet()
self.net.init_layers(2, [1], 1)
self.rec_config = RecurrentConfig()
def setUp(self):
self.net = NeuralNet()
self.net.init_layers(2, [1], 1)
self.rec_config = RecurrentConfig()
