示例#1
0
 def test_name_and_node_index(self):
     self.assertEqual(NodeSpec.from_str('abc:4'),
                      NodeSpec(layer_name='abc', node_index=4))
示例#2
0
 def test_invalid_name(self):
     with self.assertRaises(ValueError):
         NodeSpec.from_str('abc:4:3')
示例#3
0
 def test_name_only(self):
     self.assertEqual(NodeSpec.from_str('abc'),
                      NodeSpec(layer_name='abc', node_index=0))