def test_output_user_data_repack_ports(self): graph = build_graph(self.nodes, self.edges) output = output_user_data_repack(graph, ['Cc:1', '0:Cc']) self.assertDictEqual(output, {'C': [{'out': 1}, {'in': 0}]})
def test_output_user_data_repack_none(self): graph = build_graph(self.nodes, self.edges) output = output_user_data_repack(graph, None) self.assertEqual(output, None)
def test_output_user_data_repack(self): graph = build_graph(self.nodes, self.edges) output = output_user_data_repack(graph, ['Cc']) self.assertDictEqual(output, {'C': [{'port': None}]})