def setUp(self):
self.comp_holder = ComponentHolder()
self.layer1 = F.Linear(2, 2)
self.layer2 = F.Linear(2, 4)
self.comp_holder.append_comp('layer1', self.layer1)
self.comp_holder.append_comp('layer2', self.layer2)
pass
class Test_ComponentHolder(unittest.TestCase):
def setUp(self):
self.comp_holder = ComponentHolder()
self.layer1 = F.Linear(2, 2)
self.layer2 = F.Linear(2, 4)
self.comp_holder.append_comp('layer1', self.layer1)
self.comp_holder.append_comp('layer2', self.layer2)
pass
def test_append_comp(self):
self.assertIs(self.comp_holder['layer1'], self.layer1)
a, b = self.comp_holder['layer1', 'layer2']
self.assertIs(a, self.layer1)
self.assertIs(b, self.layer2)
name, comp = self.comp_holder[0]
self.assertEqual(name, 'layer1')
self.assertIs(comp, self.layer1)
def tearDown(self):
pass
