def _build_graph(self): t_start = time.time() self.graph = graph.build_graph(self.offers) t_end = time.time() logging.info("Spent {}s building the graph".format( round(t_end - t_start, 2)))
def test_build_graph(self): # Hand-crafted offers to produce a profitable path offers_with_vendor = [ Offer(league="Heist", have="Orb of Regret", want="Chaos Orb", stock=8, conversion_rate=8, contact_ign="some_guy"), Offer(league="Heist", have="Orb of Alchemy", want="Orb of Regret", stock=8, conversion_rate=1, contact_ign=VENDOR_OFFER_IGN), Offer(league="Heist", have="Chaos Orb", want="Orb of Alchemy", stock=100, conversion_rate=.25, contact_ign="some_guy"), ] graph = build_graph(offers_with_vendor) paths = find_paths(graph, "Chaos Orb", "Chaos Orb", user_config) self.assertTrue(len(paths) > 0)
def test_build_graph(self): graph = build_graph(test_offers) self.assertDictEqual(graph, expected_graph)