def get_name_and_port(tensor_name): node_name, in_port, out_port = extract_port_from_string(tensor_name) assert in_port is None or out_port is None if in_port is not None: return node_name, in_port elif out_port is not None: return node_name, out_port else: return node_name, 0
def test_no_port(self): name, in_port, out_port = extract_port_from_string('node_name') self.assertEqual(name, 'node_name') self.assertEqual(in_port, None) self.assertEqual(out_port, None)