MCPcopy Create free account
hub / github.com/pytorch/pytorch / test_createUniqueDataNode

Method test_createUniqueDataNode

caffe2/python/nomnigraph_test.py:258–271  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

256 ng.render(g)
257
258 def test_createUniqueDataNode(self):
259 net = core.Net("name")
260 nn = ng.NNModule(net)
261 n1 = nn.createUniqueDataNode("a")
262 self.assertEqual(n1.name[0], "a")
263 n2 = nn.dataFlow.createNode(ng.Operator("test1"))
264 nn.createEdge(n1, n2)
265 n3 = nn.createUniqueDataNode("a")
266 nn.createEdge(n2, n3)
267 self.assertEqual(n3.name[0], "a")
268 self.assertNotEqual(n1.name, n3.name)
269 n1 = nn.createUniqueDataNode("b")
270 n2 = nn.createUniqueDataNode("b")
271 self.assertNotEqual(n1.name, n2.name)
272
273 def test_convertToProto(self):
274 net = core.Net("name")

Callers

nothing calls this directly

Calls 6

createUniqueDataNodeMethod · 0.95
createEdgeMethod · 0.95
NetMethod · 0.80
createNodeMethod · 0.80
assertNotEqualMethod · 0.80
assertEqualMethod · 0.45

Tested by

no test coverage detected