fix tests and remove additional keyformatter calls
parent
51d1c27f2d
commit
0cfeb8621c
|
@ -415,16 +415,16 @@ TEST(DiscreteFactorGraph, DotWithNames) {
|
||||||
"graph {\n"
|
"graph {\n"
|
||||||
" size=\"5,5\";\n"
|
" size=\"5,5\";\n"
|
||||||
"\n"
|
"\n"
|
||||||
" varC[label=\"C\"];\n"
|
" var0[label=\"C\"];\n"
|
||||||
" varA[label=\"A\"];\n"
|
" var1[label=\"A\"];\n"
|
||||||
" varB[label=\"B\"];\n"
|
" var2[label=\"B\"];\n"
|
||||||
"\n"
|
"\n"
|
||||||
" factor0[label=\"\", shape=point];\n"
|
" factor0[label=\"\", shape=point];\n"
|
||||||
" varC--factor0;\n"
|
" var0--factor0;\n"
|
||||||
" varA--factor0;\n"
|
" var1--factor0;\n"
|
||||||
" factor1[label=\"\", shape=point];\n"
|
" factor1[label=\"\", shape=point];\n"
|
||||||
" varC--factor1;\n"
|
" var0--factor1;\n"
|
||||||
" varB--factor1;\n"
|
" var2--factor1;\n"
|
||||||
"}\n";
|
"}\n";
|
||||||
EXPECT(actual == expected);
|
EXPECT(actual == expected);
|
||||||
}
|
}
|
||||||
|
|
|
@ -53,8 +53,9 @@ void BayesNet<CONDITIONAL>::dot(std::ostream& os,
|
||||||
auto frontals = conditional->frontals();
|
auto frontals = conditional->frontals();
|
||||||
const Key me = frontals.front();
|
const Key me = frontals.front();
|
||||||
auto parents = conditional->parents();
|
auto parents = conditional->parents();
|
||||||
for (const Key& p : parents)
|
for (const Key& p : parents) {
|
||||||
os << " var" << keyFormatter(p) << "->var" << keyFormatter(me) << "\n";
|
os << " var" << p << "->var" << me << "\n";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
os << "}";
|
os << "}";
|
||||||
|
|
|
@ -104,16 +104,16 @@ TEST(SymbolicBayesNet, Dot) {
|
||||||
"digraph {\n"
|
"digraph {\n"
|
||||||
" size=\"5,5\";\n"
|
" size=\"5,5\";\n"
|
||||||
"\n"
|
"\n"
|
||||||
" vara1[label=\"a1\", pos=\"1,2!\", shape=box];\n"
|
" var6989586621679009793[label=\"a1\", pos=\"1,2!\", shape=box];\n"
|
||||||
" vara2[label=\"a2\", pos=\"2,2!\", shape=box];\n"
|
" var6989586621679009794[label=\"a2\", pos=\"2,2!\", shape=box];\n"
|
||||||
" varx1[label=\"x1\", pos=\"1,1!\"];\n"
|
" var8646911284551352321[label=\"x1\", pos=\"1,1!\"];\n"
|
||||||
" varx2[label=\"x2\", pos=\"2,1!\"];\n"
|
" var8646911284551352322[label=\"x2\", pos=\"2,1!\"];\n"
|
||||||
" varx3[label=\"x3\", pos=\"3,1!\"];\n"
|
" var8646911284551352323[label=\"x3\", pos=\"3,1!\"];\n"
|
||||||
"\n"
|
"\n"
|
||||||
" varx1->varx2\n"
|
" var8646911284551352321->var8646911284551352322\n"
|
||||||
" vara1->varx2\n"
|
" var6989586621679009793->var8646911284551352322\n"
|
||||||
" varx2->varx3\n"
|
" var8646911284551352322->var8646911284551352323\n"
|
||||||
" vara2->varx3\n"
|
" var6989586621679009794->var8646911284551352323\n"
|
||||||
"}");
|
"}");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -335,21 +335,21 @@ TEST(NonlinearFactorGraph, dot) {
|
||||||
"graph {\n"
|
"graph {\n"
|
||||||
" size=\"5,5\";\n"
|
" size=\"5,5\";\n"
|
||||||
"\n"
|
"\n"
|
||||||
" varl1[label=\"l1\"];\n"
|
" var7782220156096217089[label=\"l1\"];\n"
|
||||||
" varx1[label=\"x1\"];\n"
|
" var8646911284551352321[label=\"x1\"];\n"
|
||||||
" varx2[label=\"x2\"];\n"
|
" var8646911284551352322[label=\"x2\"];\n"
|
||||||
"\n"
|
"\n"
|
||||||
" factor0[label=\"\", shape=point];\n"
|
" factor0[label=\"\", shape=point];\n"
|
||||||
" varx1--factor0;\n"
|
" var8646911284551352321--factor0;\n"
|
||||||
" factor1[label=\"\", shape=point];\n"
|
" factor1[label=\"\", shape=point];\n"
|
||||||
" varx1--factor1;\n"
|
" var8646911284551352321--factor1;\n"
|
||||||
" varx2--factor1;\n"
|
" var8646911284551352322--factor1;\n"
|
||||||
" factor2[label=\"\", shape=point];\n"
|
" factor2[label=\"\", shape=point];\n"
|
||||||
" varx1--factor2;\n"
|
" var8646911284551352321--factor2;\n"
|
||||||
" varl1--factor2;\n"
|
" var7782220156096217089--factor2;\n"
|
||||||
" factor3[label=\"\", shape=point];\n"
|
" factor3[label=\"\", shape=point];\n"
|
||||||
" varx2--factor3;\n"
|
" var8646911284551352322--factor3;\n"
|
||||||
" varl1--factor3;\n"
|
" var7782220156096217089--factor3;\n"
|
||||||
"}\n";
|
"}\n";
|
||||||
|
|
||||||
const NonlinearFactorGraph fg = createNonlinearFactorGraph();
|
const NonlinearFactorGraph fg = createNonlinearFactorGraph();
|
||||||
|
@ -363,21 +363,21 @@ TEST(NonlinearFactorGraph, dot_extra) {
|
||||||
"graph {\n"
|
"graph {\n"
|
||||||
" size=\"5,5\";\n"
|
" size=\"5,5\";\n"
|
||||||
"\n"
|
"\n"
|
||||||
" varl1[label=\"l1\", pos=\"0,0!\"];\n"
|
" var7782220156096217089[label=\"l1\", pos=\"0,0!\"];\n"
|
||||||
" varx1[label=\"x1\", pos=\"1,0!\"];\n"
|
" var8646911284551352321[label=\"x1\", pos=\"1,0!\"];\n"
|
||||||
" varx2[label=\"x2\", pos=\"1,1.5!\"];\n"
|
" var8646911284551352322[label=\"x2\", pos=\"1,1.5!\"];\n"
|
||||||
"\n"
|
"\n"
|
||||||
" factor0[label=\"\", shape=point];\n"
|
" factor0[label=\"\", shape=point];\n"
|
||||||
" varx1--factor0;\n"
|
" var8646911284551352321--factor0;\n"
|
||||||
" factor1[label=\"\", shape=point];\n"
|
" factor1[label=\"\", shape=point];\n"
|
||||||
" varx1--factor1;\n"
|
" var8646911284551352321--factor1;\n"
|
||||||
" varx2--factor1;\n"
|
" var8646911284551352322--factor1;\n"
|
||||||
" factor2[label=\"\", shape=point];\n"
|
" factor2[label=\"\", shape=point];\n"
|
||||||
" varx1--factor2;\n"
|
" var8646911284551352321--factor2;\n"
|
||||||
" varl1--factor2;\n"
|
" var7782220156096217089--factor2;\n"
|
||||||
" factor3[label=\"\", shape=point];\n"
|
" factor3[label=\"\", shape=point];\n"
|
||||||
" varx2--factor3;\n"
|
" var8646911284551352322--factor3;\n"
|
||||||
" varl1--factor3;\n"
|
" var7782220156096217089--factor3;\n"
|
||||||
"}\n";
|
"}\n";
|
||||||
|
|
||||||
const NonlinearFactorGraph fg = createNonlinearFactorGraph();
|
const NonlinearFactorGraph fg = createNonlinearFactorGraph();
|
||||||
|
|
Loading…
Reference in New Issue