def test_write_node_dict(file_r_name, path_w):
# build default node_dict
with open(file_r_name, 'r') as file_json:
node_dict = lem.build_node_dict(file_json)
# default
with open(path_w + 'nodes.csv', 'w') as file_w:
lem.write_node_dict(file_w, node_dict)
# different options
path_w += 'options/'
if not os.path.exists(path_w):
os.makedirs(path_w)
# with cut
with open(path_w + 'nodes (cut=10).csv', 'w') as file_w:
lem.write_node_dict(file_w, node_dict, cut=10)
# with '\t'
with open(path_w + 'nodes.tsv', 'w') as file_w:
lem.write_node_dict(file_w, node_dict, sep='\t')
print('Done test_write_node_dict')
file_nodes_name = path_w + 'nodes.csv'
file_edges_name = path_w + 'edges.csv'
file_nodes_json_name = path_w + 'nodes.json'
file_edges_json_name = path_w + 'edges.json'
# Создаём узлы
with open(file_json_name, 'r') as file_json, \
open(file_weight_name, 'r') as file_weight, \
open(file_nodes_name, 'w') as file_nodes, \
open(file_nodes_json_name, 'w') as file_nodes_json:
# создаём словарь узлов
node_dict = lem.build_node_dict(file_json, weights=file_weight,
include_bastard=bastard,
include_non_cyrillic=non_cyrillic)
# записываем узлы в файл для Gephi
lem.write_node_dict(file_nodes, node_dict)
# отдельно сохраняем json-версию словаря узлов (для отладки)
pretty_json(node_dict, file_nodes_json)
# Создаём рёбра
with open(file_json_name, 'r') as file_json, \
open(file_weight_name, 'r') as file_weight, \
open(file_edges_name, 'w') as file_edges, \
open(file_edges_json_name, 'w') as file_edges_json:
# создаём словарь рёбер
edge_dict = lem.build_edge_dict(file_json, weights=file_weight,
include_bastard=bastard,
include_non_cyrillic=non_cyrillic)
# записываем рёбра в файл для Gephi
lem.write_edge_dict(file_edges, edge_dict)
# отдельно сохраняем json-версию словаря рёбер (для отладки)
