def test_run_fa2_layout(self):
db = self.db
graph = generate_graph(db=db, timespans_id=self.timespan['timespans_id'])
run_fa2_layout(graph=graph, memory_limit_mb=512)
positions = nx.get_node_attributes(graph, 'position')
for n in graph.nodes:
assert len(positions[n]) == 2
def test_get_display_subgraph_by_attribute(self):
db = self.db
graph = generate_graph(db=db,
timespans_id=self.timespan['timespans_id'])
num_display_nodes = 10
graph = get_display_subgraph_by_attribute(
graph=graph,
attribute='media_inlink_count',
num_nodes=num_display_nodes,
)
assert len(graph.nodes) == num_display_nodes
def test_generate_and_store_maps(self):
db = self.db
graph = generate_graph(db=db, topics_id=self.timespan['topics_id'], timespans_id=self.timespan['timespans_id'])
media_names = [{'media_id': m['media_id'], 'value': m['name']} for m in self.all_media]
attribute = {'name': 'media_name', 'data': media_names}
add_attribute_to_graph(graph, attribute)
node_media_names = nx.get_node_attributes(graph, 'media_name')
node_names = nx.get_node_attributes(graph, 'name')
log.warning(node_media_names)
log.warning(node_names)
for i in node_names:
assert node_media_names[i] == node_names[i]