def setUp(self): self.relation_manager = RelationManager() _dummy_citation_network = [{"record": 1, "references": [11, 15, 17, 18], "authors": [1, 2, 3]}, {"record": 2, "references": [15, 17, 18], "authors": [2, 3]}, {"record": 3, "references": [1, 2, 18], "authors": [3, 2]}] for citation_graph_item in _dummy_citation_network: for reference in citation_graph_item["references"]: self.relation_manager.add_reference(citation_graph_item["record"], reference) for author in citation_graph_item["authors"]: self.relation_manager.add_author(citation_graph_item["record"], author)