1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#![doc = include_str!("../README.md")]
#![cfg_attr(docsrs, feature(doc_cfg, doc_auto_cfg))]
pub mod force;
mod graph;
mod simulation;
#[cfg(feature = "json")]
pub mod json;
#[cfg(feature = "gml")]
pub mod gml;
pub mod dot;
pub use glam;
pub use petgraph;
pub use {
graph::{ForceGraph, ForceGraphHelper},
simulation::{Dimensions, Node, Simulation, SimulationParameters},
};
mod tests {
#[test]
fn dot() {
use super::{dot, ForceGraph, ForceGraphHelper};
const RESULT: &str = "graph {\n 0 [ label = \"one\" ]\n 1 [ label = \"two\" ]\n 2 [ label = \"three\" ]\n 0 -- 1 [ ]\n 1 -- 2 [ ]\n}\n";
let mut graph: ForceGraph<(), ()> = ForceGraph::default();
let one = graph.add_force_node("one", ());
let two = graph.add_force_node("two", ());
let three = graph.add_force_node("three", ());
graph.add_edge(one, two, ());
graph.add_edge(two, three, ());
let dot = dot::graph_to_dot(&graph).unwrap();
assert_eq!(dot, RESULT);
}
#[test]
#[cfg(feature = "gml")]
fn gml() {
use super::{gml, ForceGraph, ForceGraphHelper};
let mut graph: ForceGraph<(), ()> = ForceGraph::default();
let one = graph.add_force_node("one", ());
let two = graph.add_force_node("two", ());
let three = graph.add_force_node("three", ());
graph.add_edge(one, two, ());
graph.add_edge(two, three, ());
let gml = gml::graph_to_gml(&graph);
let new_graph = gml::graph_from_gml(gml).unwrap();
assert_eq!(new_graph.node_count(), 3);
assert_eq!(new_graph.edge_count(), 2);
assert!(new_graph
.node_weights()
.find(|x| x.name == "one".to_string())
.is_some());
assert!(new_graph
.node_weights()
.find(|x| x.name == "two".to_string())
.is_some());
assert!(new_graph
.node_weights()
.find(|x| x.name == "three".to_string())
.is_some());
}
#[test]
#[cfg(feature = "json")]
fn json() {
use super::{json, ForceGraph, ForceGraphHelper};
let mut graph: ForceGraph<&str, &str> = ForceGraph::default();
let one = graph.add_force_node("one", "onedata");
let two = graph.add_force_node("two", "twodata");
let three = graph.add_force_node("three", "threedata");
graph.add_edge(one, two, "onetwoedgedata");
graph.add_edge(two, three, "twothreeedgedata");
let json = json::graph_to_json(&graph).unwrap();
let ag = json::graph_from_json(json.to_string()).unwrap();
assert_eq!(ag.node_count(), 3);
assert_eq!(ag.edge_count(), 2);
assert_eq!(
ag.node_weights()
.find(|x| x.name == "one")
.unwrap()
.data
.get("metadata")
.unwrap()
.to_string()
.replace('"', ""),
"onedata".to_string()
);
assert!(ag
.edge_weights()
.find(|x| x.to_string().replace('"', "") == "onetwoedgedata".to_string())
.is_some());
}
}