@Test
public void test4() {
// For X3
Map<String, String[]> templates = new HashMap<>();
templates.put(
"NEW(b) + NEW(b) + NEW(c) + NEW(c) + NEW(c)", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("NEW(X1) + NEW(b) + NEW(c) + NEW(c) + NEW(c)", new String[] {});
templates.put("$", new String[] {});
templates.put("TSUM($)", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("TPROD($)", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("TPROD($) + X2", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("TPROD($) + TSUM($)", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("tan(TSUM(NEW(a)*$))", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("Normal(0, 1)", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("Normal(m, s)", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("Normal(NEW(m), s)", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("Normal(NEW(m), NEW(s)) + m1 + s6", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("TSUM($) + a", new String[] {"X1", "X2", "X3", "X4", "X5"});
templates.put("TSUM($) + TSUM($) + TSUM($) + 1", new String[] {"X1", "X2", "X3", "X4", "X5"});
for (String template : templates.keySet()) {
GeneralizedSemPm semPm = makeTypicalPm();
print(semPm.getGraph().toString());
Set<Node> shouldWork = new HashSet<>();
for (String name : templates.get(template)) {
shouldWork.add(semPm.getNode(name));
}
Set<Node> works = new HashSet<>();
for (int i = 0; i < semPm.getNodes().size(); i++) {
print("-----------");
print(semPm.getNodes().get(i).toString());
print("Trying template: " + template);
String _template = template;
Node node = semPm.getNodes().get(i);
try {
_template = TemplateExpander.getInstance().expandTemplate(_template, semPm, node);
} catch (Exception e) {
print("Couldn't expand template: " + template);
continue;
}
try {
semPm.setNodeExpression(node, _template);
print("Set formula " + _template + " for " + node);
if (semPm.getVariableNodes().contains(node)) {
works.add(node);
}
} catch (Exception e) {
print("Couldn't set formula " + _template + " for " + node);
}
}
for (String parameter : semPm.getParameters()) {
print("-----------");
print(parameter);
print("Trying template: " + template);
String _template = template;
try {
_template = TemplateExpander.getInstance().expandTemplate(_template, semPm, null);
} catch (Exception e) {
print("Couldn't expand template: " + template);
continue;
}
try {
semPm.setParameterExpression(parameter, _template);
print("Set formula " + _template + " for " + parameter);
} catch (Exception e) {
print("Couldn't set formula " + _template + " for " + parameter);
}
}
assertEquals(shouldWork, works);
}
}
