def test_post_paraphrase_cleanup_exclude_unrend(self):
fake_graph = graph.WikiMap()
fake_graph.add_mapping("foo", "A", "B", False)
fake_graph.add_mapping("foo", "B", "C", False)
fake_graph.add_mapping("foo", "B", "D", False)
fake_graph.add_mapping("junk", "B", "C", False)
self.assertItemsEqual(
synonyms.post_paraphrase_cleanup(["A", "B", "C"], True,
fake_graph), ["B", "C"])
def setUp(self):
self.G = graph.WikiMap()
# four node group
self.G.add_edge("A", "B")
self.G.add_edge("A", "C")
self.G.add_edge("D", "C")
# three node group
self.G.add_edge("Y", "X")
self.G.add_edge("Z", "X")
# three node group (another)
self.G.add_edge("alpha", "beta")
self.G.add_edge("alpha", "gamma")
# two node group
self.G.add_edge("M", "N")
# EXPECTED:
# three node group
self.expected_three1 = graph.WikiMap()
self.expected_three1.add_edge("Y", "X")
self.expected_three1.add_edge("X", "Y")
self.expected_three1.add_edge("Z", "X")
self.expected_three1.add_edge("X", "Z")
# three node group (another)
self.expected_three2 = graph.WikiMap()
self.expected_three2.add_edge("alpha", "beta")
self.expected_three2.add_edge("beta", "alpha")
self.expected_three2.add_edge("alpha", "gamma")
self.expected_three2.add_edge("gamma", "alpha")
# four node group
self.expected_four = graph.WikiMap()
self.expected_four.add_edge("A", "B")
self.expected_four.add_edge("B", "A")
self.expected_four.add_edge("A", "C")
self.expected_four.add_edge("C", "A")
self.expected_four.add_edge("D", "C")
self.expected_four.add_edge("C", "D")
def create_graph(infoboxes, clean):
"""Creates and saves a graph to [graph_path] from excel file located at
[excel_path], optionally [clean]ing (boolean) the graph
"""
G = graph.WikiMap()
for infobox in infoboxes:
for unrend, rend in data.get_single_mappings(infobox).iteritems():
G.add_mapping(infobox, unrend, rend, clean)
return G
def setUp(self):
self.fake_graph = graph.WikiMap()
self.fake_graph.add_mapping("random", "A", "X", False)
self.fake_graph.add_mapping("bar", "F", "A", False)
self.fake_graph.add_mapping("bar", "E", "A", False)
self.fake_graph.add_mapping("foo", "E", "A", False)
self.fake_graph.add_mapping("foo", "A", "B", False)
self.fake_graph.add_mapping("bar", "E", "D", False)
self.fake_graph.add_mapping("junk", "E", "D", False)
self.fake_graph.add_mapping("foo", "B", "D", False)
self.fake_graph.add_mapping("foo", "B", "C", False)
self.fake_graph.add_mapping("junk", "B", "C", False)
patcher = mock.patch('wikimap.synonyms.similar_enough')
self.mock = patcher.start(
) # patcher.start() RETURNS the mocked object
self.mock.side_effect = lambda x, y: x == y
self.addCleanup(patcher.stop)
def setUp(self):
self.G = graph.WikiMap()
self.G.add_edge('unrend', 'hybrid')
self.G.add_edge('hybrid', 'rend')
self.G.node['unrend']['was'] = ['_unrend_']
self.G.node['hybrid']['was'] = ['#hybrid#', '_hybrid_']
self.G.node['rend']['was'] = ['_rend_']
self.G.node['unrend']['infobox'] = {'Infobox foo bar': ['unrend']}
self.G.node['hybrid']['infobox'] = {
'Infobox foo bar': ['rend'],
'Infobox baz bang': ['unrend', 'rend']
}
self.G.node['rend']['infobox'] = {'Infobox baz bang': ['rend']}
self.G.edge['unrend']['hybrid']['infobox'] = ['Infobox foo bar']
self.G.edge['hybrid']['rend']['infobox'] = [
'Infobox baz bang', 'Infobox aba zaba'
]
def test_connected_component_nodes_with_size(self):
G = graph.WikiMap()
# four node group
G.add_edge("A", "B")
G.add_edge("A", "C")
G.add_edge("D", "C")
# three node group
G.add_edge("Y", "X")
G.add_edge("Z", "X")
# three node group (another)
G.add_edge("alpha", "beta")
G.add_edge("alpha", "gamma")
# two node group
G.add_edge("M", "N")
self.assertItemsEqual(G.connected_component_nodes_with_size(3),
[["Y", "X", "Z"], ["alpha", "beta", "gamma"]])
def setUp(self):
self.G = graph.WikiMap()