def setUp(self, mock_load_config, mock_delete_all, mock_open, mock_multidigraph): mock_open.return_value = "config.yaml" self.base_graph = dict(graph_db='graph_db', refresh=True) self.statuses = dict(status=['Status1', 'Status2'], motd='I am the MOTD') def load_config(filename): if "config.yaml" in filename: return self.base_graph elif "statuses.yaml" in filename: return self.statuses mock_load_config.side_effect = load_config self.popittoneo = PopItToNeo() self.graph_patcher = patch("popit_to_networkx.nx.MultiDiGraph") self.add_node_patcher = patch( "popit_to_networkx.nx.MultiDiGraph.PopItRelationship.add_node") self.graph = self.graph_patcher.start() self.add_node = self.add_node_patcher.start() self.popit2networkx = PopItRelationship()