public void testConcurrentModification() {
Graph graph = graphTest.generateGraph();
Vertex a = graph.addVertex(null);
graph.addVertex(null);
graph.addVertex(null);
if (graph.getFeatures().supportsVertexIteration) {
for (Vertex vertex : graph.getVertices()) {
graph.addEdge(null, vertex, a, convertId(graph, "x"));
graph.addEdge(null, vertex, a, convertId(graph, "y"));
}
for (Vertex vertex : graph.getVertices()) {
assertEquals(BaseTest.count(vertex.getEdges(Direction.OUT)), 2);
for (Edge edge : vertex.getEdges(Direction.OUT)) {
graph.removeEdge(edge);
}
}
for (Vertex vertex : graph.getVertices()) {
graph.removeVertex(vertex);
}
} else if (graph.getFeatures().supportsEdgeIteration) {
for (int i = 0; i < 10; i++) {
graph.addEdge(null, graph.addVertex(null), graph.addVertex(null), convertId(graph, "test"));
}
for (Edge edge : graph.getEdges()) {
graph.removeEdge(edge);
}
}
graph.shutdown();
}
public void testAddingVerticesAndEdges() {
Graph graph = graphTest.generateGraph();
Vertex a = graph.addVertex(null);
Vertex b = graph.addVertex(null);
Edge edge = graph.addEdge(null, a, b, convertId(graph, "knows"));
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(1, count(graph.getEdges()));
}
if (graph.getFeatures().supportsVertexIteration) {
assertEquals(2, count(graph.getVertices()));
}
graph.removeVertex(a);
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(0, count(graph.getEdges()));
}
if (graph.getFeatures().supportsVertexIteration) {
assertEquals(1, count(graph.getVertices()));
}
try {
graph.removeEdge(edge);
// TODO: doesn't work with wrapper graphs assertTrue(false);
} catch (Exception e) {
assertTrue(true);
}
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(0, count(graph.getEdges()));
}
if (graph.getFeatures().supportsVertexIteration) {
assertEquals(1, count(graph.getVertices()));
}
graph.shutdown();
}
/** Get a graph and direct only the unshielded colliders. */
public static void basicPattern(Graph graph) {
Set<Edge> undirectedEdges = new HashSet<Edge>();
NEXT_EDGE:
for (Edge edge : graph.getEdges()) {
Node head = null, tail = null;
if (edge.getEndpoint1() == Endpoint.ARROW && edge.getEndpoint2() == Endpoint.TAIL) {
head = edge.getNode1();
tail = edge.getNode2();
} else if (edge.getEndpoint2() == Endpoint.ARROW && edge.getEndpoint1() == Endpoint.TAIL) {
head = edge.getNode2();
tail = edge.getNode1();
}
if (head != null) {
for (Node node : graph.getParents(head)) {
if (node != tail && !graph.isAdjacentTo(tail, node)) {
continue NEXT_EDGE;
}
}
undirectedEdges.add(edge);
}
}
for (Edge nextUndirected : undirectedEdges) {
Node node1 = nextUndirected.getNode1(), node2 = nextUndirected.getNode2();
graph.removeEdge(nextUndirected);
graph.addUndirectedEdge(node1, node2);
}
}
// driver
public static void main(String[] args) {
// create empty graph with 5 vertices
Graph myGraph = new Graph(5);
// myGraph.printMatrix();
// addEdge: D->C:10, A->B:12, D->B:23, A->D:87, E->D:43, B->E:11, C->A:19
myGraph.addEdge("D", "C", 10);
myGraph.addEdge("A", "B", 12);
myGraph.addEdge("D", "B", 23);
myGraph.addEdge("A", "D", 87);
myGraph.addEdge("E", "D", 43);
myGraph.addEdge("B", "E", 11);
myGraph.addEdge("C", "A", 19);
myGraph.removeEdge("A", "D", 87);
myGraph.changeWeight("A", "B", 13);
// myGraph.printMatrix();
myGraph.bfs("D");
System.out.println("Finished");
}
public static Graph weightedRandomGraph(int n, int e) {
List<Node> nodes = new ArrayList<Node>();
for (int i = 0; i < n; i++) nodes.add(new GraphNode("X" + i));
Graph graph = new EdgeListGraph(nodes);
for (int e0 = 0; e0 < e; e0++) {
int i1 = weightedRandom(nodes, graph);
// int i2 = RandomUtil.getInstance().nextInt(n);
int i2 = weightedRandom(nodes, graph);
if (!(shortestPath(nodes.get(i1), nodes.get(i2), graph) < 9)) {
e0--;
continue;
}
if (i1 == i2) {
e0--;
continue;
}
Edge edge = Edges.undirectedEdge(nodes.get(i1), nodes.get(i2));
if (graph.containsEdge(edge)) {
e0--;
continue;
}
graph.addEdge(edge);
}
for (Edge edge : graph.getEdges()) {
Node n1 = edge.getNode1();
Node n2 = edge.getNode2();
if (!graph.isAncestorOf(n2, n1)) {
graph.removeEdge(edge);
graph.addDirectedEdge(n1, n2);
} else {
graph.removeEdge(edge);
graph.addDirectedEdge(n2, n1);
}
}
return graph;
}
private void calcStats() {
// Graph resultGraph = getAlgorithmRunner().getResultGraph();
IGesRunner runner = (IGesRunner) getAlgorithmRunner();
if (runner.getTopGraphs().isEmpty()) {
throw new IllegalArgumentException(
"No patterns were recorded. Please adjust the number of " + "patterns to store.");
}
Graph resultGraph = runner.getTopGraphs().get(runner.getIndex()).getGraph();
if (getAlgorithmRunner().getDataModel() instanceof DataSet) {
// resultGraph may be the output of a PC search.
// Such graphs sometimes contain doubly directed edges.
// /We converte such edges to directed edges here.
// For the time being an orientation is arbitrarily selected.
Set<Edge> allEdges = resultGraph.getEdges();
for (Edge edge : allEdges) {
if (edge.getEndpoint1() == Endpoint.ARROW && edge.getEndpoint2() == Endpoint.ARROW) {
// Option 1 orient it from node1 to node2
resultGraph.setEndpoint(edge.getNode1(), edge.getNode2(), Endpoint.ARROW);
// Option 2 remove such edges:
resultGraph.removeEdge(edge);
}
}
Pattern pattern = new Pattern(resultGraph);
PatternToDag ptd = new PatternToDag(pattern);
Graph dag = ptd.patternToDagMeekRules();
DataSet dataSet = (DataSet) getAlgorithmRunner().getDataModel();
String report;
if (dataSet.isContinuous()) {
report = reportIfContinuous(dag, dataSet);
} else if (dataSet.isDiscrete()) {
report = reportIfDiscrete(dag, dataSet);
} else {
throw new IllegalArgumentException("");
}
JScrollPane dagWorkbenchScroll = dagWorkbenchScroll(dag);
modelStatsText.setLineWrap(true);
modelStatsText.setWrapStyleWord(true);
modelStatsText.setText(report);
removeStatsTabs();
tabbedPane.addTab("DAG in pattern", dagWorkbenchScroll);
tabbedPane.addTab("DAG Model Statistics", modelStatsText);
}
}
public static Graph bestGuessCycleOrientation(Graph graph, IndependenceTest test) {
while (true) {
List<Node> cycle = GraphUtils.directedCycle(graph);
if (cycle == null) {
break;
}
LinkedList<Node> _cycle = new LinkedList<Node>(cycle);
Node first = _cycle.getFirst();
Node last = _cycle.getLast();
_cycle.addFirst(last);
_cycle.addLast(first);
int _j = -1;
double minP = Double.POSITIVE_INFINITY;
for (int j = 1; j < _cycle.size() - 1; j++) {
int i = j - 1;
int k = j + 1;
Node x = test.getVariable(_cycle.get(i).getName());
Node y = test.getVariable(_cycle.get(j).getName());
Node z = test.getVariable(_cycle.get(k).getName());
test.isIndependent(x, z, Collections.singletonList(y));
System.out.println("Testing " + x + " _||_ " + z + " | " + y);
double p = test.getPValue();
System.out.println("p = " + p);
if (p < minP) {
_j = j;
minP = p;
}
}
Node x = _cycle.get(_j - 1);
Node y = _cycle.get(_j);
Node z = _cycle.get(_j + 1);
graph.removeEdge(x, y);
graph.removeEdge(z, y);
graph.addDirectedEdge(x, y);
graph.addDirectedEdge(z, y);
}
return graph;
}
public void testRemoveManyEdges() {
Graph graph = graphTest.getGraphInstance();
long counter = 200000l;
int edgeCount = 100;
Set<Edge> edges = new HashSet<Edge>();
for (int i = 0; i < edgeCount; i++) {
Vertex out = graph.addVertex(convertId("" + counter++));
Vertex in = graph.addVertex(convertId("" + counter++));
edges.add(graph.addEdge(null, out, in, convertId("a" + UUID.randomUUID().toString())));
}
assertEquals(edgeCount, edges.size());
if (graphTest.supportsVertexIteration) {
this.stopWatch();
assertEquals(edgeCount * 2, count(graph.getVertices()));
BaseTest.printPerformance(
graph.toString(), edgeCount * 2, "vertices counted", this.stopWatch());
}
if (graphTest.supportsEdgeIteration) {
this.stopWatch();
assertEquals(edgeCount, count(graph.getEdges()));
BaseTest.printPerformance(graph.toString(), edgeCount, "edges counted", this.stopWatch());
int i = edgeCount;
this.stopWatch();
for (Edge edge : edges) {
graph.removeEdge(edge);
i--;
assertEquals(i, count(graph.getEdges()));
if (graphTest.supportsVertexIteration) {
int x = 0;
for (Vertex vertex : graph.getVertices()) {
if (count(vertex.getOutEdges()) > 0) {
assertEquals(1, count(vertex.getOutEdges()));
assertFalse(count(vertex.getInEdges()) > 0);
} else if (count(vertex.getInEdges()) > 0) {
assertEquals(1, count(vertex.getInEdges()));
assertFalse(count(vertex.getOutEdges()) > 0);
} else {
x++;
}
}
assertEquals((edgeCount - i) * 2, x);
}
}
BaseTest.printPerformance(
graph.toString(), edgeCount, "edges removed and graph checked", this.stopWatch());
}
graph.shutdown();
}
/** Orients according to background knowledge. */
public static void pcOrientbk(Knowledge bk, Graph graph, List<Node> nodes) {
TetradLogger.getInstance().log("info", "Staring BK Orientation.");
for (Iterator<KnowledgeEdge> it = bk.forbiddenEdgesIterator(); it.hasNext(); ) {
KnowledgeEdge edge = it.next();
// match strings to variables in the graph.
Node from = translate(edge.getFrom(), nodes);
Node to = translate(edge.getTo(), nodes);
if (from == null || to == null) {
continue;
}
if (graph.getEdge(from, to) == null) {
continue;
}
// Orient to-->from
graph.removeEdge(from, to);
graph.addDirectedEdge(from, to);
graph.setEndpoint(from, to, Endpoint.TAIL);
graph.setEndpoint(to, from, Endpoint.ARROW);
TetradLogger.getInstance()
.edgeOriented(SearchLogUtils.edgeOrientedMsg("Knowledge", graph.getEdge(to, from)));
}
for (Iterator<KnowledgeEdge> it = bk.requiredEdgesIterator(); it.hasNext(); ) {
KnowledgeEdge edge = it.next();
// match strings to variables in this graph
Node from = translate(edge.getFrom(), nodes);
Node to = translate(edge.getTo(), nodes);
if (from == null || to == null) {
continue;
}
if (graph.getEdge(from, to) == null) {
continue;
}
// Orient from-->to
graph.setEndpoint(to, from, Endpoint.TAIL);
graph.setEndpoint(from, to, Endpoint.ARROW);
TetradLogger.getInstance()
.edgeOriented(SearchLogUtils.edgeOrientedMsg("Knowledge", graph.getEdge(from, to)));
}
TetradLogger.getInstance().log("info", "Finishing BK Orientation.");
}
private void uncorrelationExogenousVariables() {
Graph graph = getWorkbench().getGraph();
Set<Edge> edges = graph.getEdges();
for (Edge edge : edges) {
if (Edges.isBidirectedEdge(edge)) {
try {
graph.removeEdge(edge);
} catch (Exception e) {
// Ignore.
}
}
}
}
public void testRemoveSelfLoops() {
Graph graph = graphTest.getGraphInstance();
if (graphTest.allowsSelfLoops) {
List<String> ids = generateIds(3);
Vertex v1 = graph.addVertex(convertId(ids.get(0)));
Vertex v2 = graph.addVertex(convertId(ids.get(1)));
Vertex v3 = graph.addVertex(convertId(ids.get(2)));
Edge e1 = graph.addEdge(null, v1, v1, convertId("is_self"));
Edge e2 = graph.addEdge(null, v2, v2, convertId("is_self"));
Edge e3 = graph.addEdge(null, v3, v3, convertId("is_self"));
if (graphTest.supportsVertexIteration) assertEquals(3, count(graph.getVertices()));
if (graphTest.supportsEdgeIteration) {
assertEquals(3, count(graph.getEdges()));
for (Edge edge : graph.getEdges()) {
assertEquals(edge.getInVertex(), edge.getOutVertex());
assertEquals(edge.getInVertex().getId(), edge.getOutVertex().getId());
}
}
graph.removeVertex(v1);
if (graphTest.supportsEdgeIteration) {
assertEquals(2, count(graph.getEdges()));
for (Edge edge : graph.getEdges()) {
assertEquals(edge.getInVertex(), edge.getOutVertex());
assertEquals(edge.getInVertex().getId(), edge.getOutVertex().getId());
}
}
assertEquals(1, count(v2.getOutEdges()));
assertEquals(1, count(v2.getInEdges()));
graph.removeEdge(e2);
assertEquals(0, count(v2.getOutEdges()));
assertEquals(0, count(v2.getInEdges()));
if (graphTest.supportsEdgeIteration) {
assertEquals(count(graph.getEdges()), 1);
for (Edge edge : graph.getEdges()) {
assertEquals(edge.getInVertex(), edge.getOutVertex());
assertEquals(edge.getInVertex().getId(), edge.getOutVertex().getId());
}
}
}
graph.shutdown();
}
public void testNoConcurrentModificationException() {
Graph graph = graphTest.generateGraph();
for (int i = 0; i < 25; i++) {
graph.addEdge(null, graph.addVertex(null), graph.addVertex(null), convertId(graph, "test"));
}
if (graph.getFeatures().supportsVertexIteration)
assertEquals(BaseTest.count(graph.getVertices()), 50);
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(BaseTest.count(graph.getEdges()), 25);
for (final Edge edge : graph.getEdges()) {
graph.removeEdge(edge);
}
assertEquals(BaseTest.count(graph.getEdges()), 0);
}
if (graph.getFeatures().supportsVertexIteration)
assertEquals(BaseTest.count(graph.getVertices()), 50);
graph.shutdown();
}
public void testRemoveSelfLoops() {
Graph graph = graphTest.generateGraph();
if (graph.getFeatures().supportsSelfLoops) {
Vertex v1 = graph.addVertex(convertId(graph, "1"));
Vertex v2 = graph.addVertex(convertId(graph, "2"));
Vertex v3 = graph.addVertex(convertId(graph, "3"));
Edge e1 = graph.addEdge(null, v1, v1, convertId(graph, "is_self"));
Edge e2 = graph.addEdge(null, v2, v2, convertId(graph, "is_self"));
Edge e3 = graph.addEdge(null, v3, v3, convertId(graph, "is_self"));
if (graph.getFeatures().supportsVertexIteration) assertEquals(3, count(graph.getVertices()));
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(3, count(graph.getEdges()));
for (Edge edge : graph.getEdges()) {
assertEquals(edge.getVertex(Direction.IN), edge.getVertex(Direction.OUT));
assertEquals(edge.getVertex(Direction.IN).getId(), edge.getVertex(Direction.OUT).getId());
}
}
graph.removeVertex(v1);
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(2, count(graph.getEdges()));
for (Edge edge : graph.getEdges()) {
assertEquals(edge.getVertex(Direction.IN), edge.getVertex(Direction.OUT));
assertEquals(edge.getVertex(Direction.IN).getId(), edge.getVertex(Direction.OUT).getId());
}
}
assertEquals(1, count(v2.getEdges(Direction.OUT)));
assertEquals(1, count(v2.getEdges(Direction.IN)));
graph.removeEdge(e2);
assertEquals(0, count(v2.getEdges(Direction.OUT)));
assertEquals(0, count(v2.getEdges(Direction.IN)));
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(count(graph.getEdges()), 1);
for (Edge edge : graph.getEdges()) {
assertEquals(edge.getVertex(Direction.IN), edge.getVertex(Direction.OUT));
assertEquals(edge.getVertex(Direction.IN).getId(), edge.getVertex(Direction.OUT).getId());
}
}
}
graph.shutdown();
}
public void testRemovingEdges() {
Graph graph = graphTest.generateGraph();
int vertexCount = 100;
int edgeCount = 200;
List<Vertex> vertices = new ArrayList<Vertex>();
List<Edge> edges = new ArrayList<Edge>();
Random random = new Random();
this.stopWatch();
for (int i = 0; i < vertexCount; i++) {
vertices.add(graph.addVertex(null));
}
printPerformance(graph.toString(), vertexCount, "vertices added", this.stopWatch());
this.stopWatch();
for (int i = 0; i < edgeCount; i++) {
Vertex a = vertices.get(random.nextInt(vertices.size()));
Vertex b = vertices.get(random.nextInt(vertices.size()));
if (a != b) {
edges.add(graph.addEdge(null, a, b, convertId(graph, "a" + UUID.randomUUID())));
}
}
printPerformance(graph.toString(), edgeCount, "edges added", this.stopWatch());
this.stopWatch();
int counter = 0;
for (Edge e : edges) {
counter = counter + 1;
graph.removeEdge(e);
if (graph.getFeatures().supportsEdgeIteration) {
assertEquals(edges.size() - counter, count(graph.getEdges()));
}
if (graph.getFeatures().supportsVertexIteration) {
assertEquals(vertices.size(), count(graph.getVertices()));
}
}
printPerformance(
graph.toString(),
edgeCount,
"edges deleted (with size check on each delete)",
this.stopWatch());
graph.shutdown();
}
private Graph pickDag(Graph graph) {
SearchGraphUtils.basicPattern(graph, false);
addRequiredEdges(graph);
boolean containsUndirected;
do {
containsUndirected = false;
for (Edge edge : graph.getEdges()) {
if (Edges.isUndirectedEdge(edge)) {
containsUndirected = true;
graph.removeEdge(edge);
Edge _edge = Edges.directedEdge(edge.getNode1(), edge.getNode2());
graph.addEdge(_edge);
}
}
meekOrient(graph, getKnowledge());
} while (containsUndirected);
return graph;
}
void removeEdge(E jtEdge) {
jtElementsBeingRemoved.add(jtEdge);
graph.removeEdge(jtEdge);
jtElementsBeingRemoved.remove(jtEdge);
}
public static void main(String[] args) {
System.out.println("Start Graph Tests");
System.out.println("Test 1 START: Basic Node and edge methods");
// Graph creation and deletion
Graph g1 = new Graph();
Node nX = new Node("nY");
Node nY = new Node("nX");
if (!g1.addNode(nX)) {
System.out.println("Test1.0.1 Fail");
}
if (!g1.addNode(nY)) {
System.out.println("Test1.0.2 Fail");
}
if (g1.addNode(nY)) {
System.out.println("Test1.0.3 Fail");
}
Edge eX = new Edge(nX, nY);
if (!g1.addEdge(eX)) {
System.out.println("Test1.0.4 Fail");
}
if (g1.addEdge(eX)) {
System.out.println("Test1.0.5 Fail");
}
// test the consistency checker
if (!g1.consistent()) {
System.out.println("Test1.0.6 Fail");
}
ArrayList<Edge> nYTo = nY.getEdgesTo();
nYTo.remove(eX);
if (g1.consistent()) {
System.out.println("Test1.0.7 Fail");
}
nYTo.add(eX);
if (!g1.consistent()) {
System.out.println("Test1.0.8 Fail");
}
ArrayList<Edge> g1Edges = g1.getEdges();
g1Edges.remove(eX);
if (g1.consistent()) {
System.out.println("Test1.0.9 Fail");
}
g1Edges.add(eX);
if (!g1.consistent()) {
System.out.println("Test1.0.10 Fail");
}
Node nOutside = new Node();
eX.setFrom(nOutside);
if (g1.consistent()) {
System.out.println("Test1.0.11 Fail");
}
eX.setFrom(nX);
if (!g1.consistent()) {
System.out.println("Test1.0.12 Fail");
}
g1 = new Graph("G");
if (!g1.getLabel().equals("G")) {
System.out.println("Test1.0.13 Fail");
}
g1.setLabel("H");
if (!g1.getLabel().equals("H")) {
System.out.println("Test1.0.14 Fail");
}
if (!g1.consistent()) {
System.out.println("Test1.0.15 Fail");
}
/* first real test graph
^e9v
n1 e1> n2 e2> n3 n6
e3v e4^ e5ve6ve7^
---n4--- <e8 n5
*/
// Graph Object creation
g1.clear();
Node n1 = new Node("n1");
if (!g1.addNode(n1)) {
System.out.println("Test1.1.1 Fail");
}
Node n2 = new Node("n2");
if (!g1.addNode(n2)) {
System.out.println("Test1.1.2 Fail");
}
Node n3 = new Node();
if (!g1.addNode(n3)) {
System.out.println("Test1.1.3 Fail");
}
Node n4 = new Node("n4");
if (!g1.addNode(n4)) {
System.out.println("Test1.1.4 Fail");
}
Node n5 = new Node("test");
if (!g1.addNode(n5)) {
System.out.println("Test1.1.5 Fail");
}
Node n6 = new Node("n6");
if (!g1.addNode(n6)) {
System.out.println("Test1.1.6 Fail");
}
Edge e1 = new Edge(n6, n6);
if (!g1.addEdge(e1)) {
System.out.println("Test1.1.7 Fail");
}
Edge e2 = new Edge(n2, n3);
if (!g1.addEdge(e2)) {
System.out.println("Test1.1.8 Fail");
}
Edge e3 = new Edge(n1, n4);
if (!g1.addEdge(e3)) {
System.out.println("Test1.1.9 Fail");
}
Edge e4 = new Edge(n4, n2);
if (!g1.addEdge(e4)) {
System.out.println("Test1.1.10 Fail");
}
Edge e5 = new Edge(n3, n5);
if (!g1.addEdge(e5)) {
System.out.println("Test1.1.11 Fail");
}
Edge e6 = new Edge(n3, n5);
if (!g1.addEdge(e6)) {
System.out.println("Test1.1.12 Fail");
}
Edge e7 = new Edge(n3, n5);
if (!g1.addEdge(e7)) {
System.out.println("Test1.1.13 Fail");
}
Edge e8 = new Edge(n1, n1);
if (!g1.addEdge(e8)) {
System.out.println("Test1.1.14 Fail");
}
Edge e9 = new Edge(n3, n3);
if (!g1.addEdge(e9)) {
System.out.println("Test1.1.15 Fail");
}
// oposite end test
if (e1.getOppositeEnd(n6) != n6) {
System.out.println("Test1.1.16 Fail");
}
if (e2.getOppositeEnd(n2) != n3) {
System.out.println("Test1.1.17 Fail");
}
if (e2.getOppositeEnd(n3) != n2) {
System.out.println("Test1.1.18 Fail");
}
if (e2.getOppositeEnd(n6) != null) {
System.out.println("Test1.1.19 Fail");
}
// changing the connections of edges
e1.setFromTo(n4, n4);
e1.setFrom(n5);
e1.setTo(n5);
e1.setFromTo(n3, n6);
e1.setFromTo(n1, n2);
e8.setFrom(n5);
e8.setTo(n4);
e7.reverse();
e9.reverse();
// resetting labels
n3.setLabel("n3");
n5.setLabel("n5");
Node tempTo = e1.getTo();
Node tempFrom = e1.getFrom();
e1.reverse();
if (e1.getTo() != tempFrom) {
System.out.println("Test1.1.20 Fail");
}
if (e1.getFrom() != tempTo) {
System.out.println("Test1.1.21 Fail");
}
e1.reverse();
if (e1.getTo() != tempTo) {
System.out.println("Test1.1.22 Fail");
}
if (e1.getFrom() != tempFrom) {
System.out.println("Test1.1.23 Fail");
}
if (!g1.consistent()) {
System.out.println("Test1.1.24 Fail");
}
// Node connectivity testing
HashSet<Node> testN = new HashSet<Node>();
if (!n6.connectingNodes().equals(testN)) {
System.out.println("Test1.2.1 Fail");
}
if (!n6.unvisitedConnectingNodes().equals(testN)) {
System.out.println("Test1.2.2 Fail");
}
if (!n6.connectingEdges().equals(testN)) {
System.out.println("Test1.2.4 Fail");
}
testN = new HashSet<Node>();
testN.add(n5);
testN.add(n3);
testN.add(n2);
if (!n3.connectingNodes().equals(testN)) {
System.out.println("Test1.2.5 Fail");
}
if (!n3.unvisitedConnectingNodes().equals(testN)) {
System.out.println("Test1.2.7 Fail");
}
// Node visited flag testing
n2.setVisited(true);
if (!n3.connectingNodes().equals(testN)) {
System.out.println("Test1.3.1 Fail");
}
testN.remove(n2);
if (!n3.unvisitedConnectingNodes().equals(testN)) {
System.out.println("Test1.3.3 Fail");
}
g1.setNodesVisited(true);
testN = new HashSet<Node>();
if (!n3.unvisitedConnectingNodes().equals(testN)) {
System.out.println("Test1.3.4 Fail");
}
n2.setVisited(false);
testN.add(n2);
if (!n3.unvisitedConnectingNodes().equals(testN)) {
System.out.println("Test1.3.5 Fail");
}
testN.remove(n2);
n2.setVisited(true);
testN = new HashSet<Node>();
if (!n3.unvisitedConnectingNodes().equals(testN)) {
System.out.println("Test1.3.6 Fail");
}
testN = new HashSet<Node>();
testN.add(n5);
testN.add(n3);
testN.add(n2);
if (!n3.connectingNodes().equals(testN)) {
System.out.println("Test1.3.7 Fail");
}
g1.setNodesVisited();
// Edge visited flag testing
HashSet<Edge> testE = new HashSet<Edge>();
testE = new HashSet<Edge>();
testE.add(e3);
testE.add(e1);
if (!n1.connectingEdges().equals(testE)) {
System.out.println("Test1.5.1 Fail");
}
if (!n1.unvisitedConnectingEdges().equals(testE)) {
System.out.println("Test1.5.2 Fail");
}
testE.remove(e1);
e1.setVisited(true);
if (!n1.unvisitedConnectingEdges().equals(testE)) {
System.out.println("Test1.5.3 Fail");
}
e1.setVisited(false);
e2.setVisited(false);
g1.setEdgesVisited(true);
testE = new HashSet<Edge>();
if (!n1.unvisitedConnectingEdges().equals(testE)) {
System.out.println("Test1.5.4 Fail");
}
e1.setVisited(true);
e2.setVisited(true);
g1.setEdgesVisited(false);
testE = new HashSet<Edge>();
testE.add(e3);
testE.add(e1);
if (!n1.connectingEdges().equals(testE)) {
System.out.println("Test1.5.5 Fail");
}
if (!n1.unvisitedConnectingEdges().equals(testE)) {
System.out.println("Test1.5.6 Fail");
}
System.out.print("Test 1 END");
// test dynamic graph stuff
System.out.println(" | Test 2 START: Object removal");
Node nd1 = new Node("nd1");
Node nd2 = new Node("nd2");
Node nd3 = new Node("nd3");
Graph g2 = new Graph("g2");
g2.addNode(nd1);
g2.addNode(nd2);
g2.addNode(nd3);
Edge ed1 = new Edge(nd1, nd2);
Edge ed2 = new Edge(nd1, nd2);
Edge ed3 = new Edge(nd2, nd1);
Edge ed4 = new Edge(nd1, nd1);
Edge ed5 = new Edge(nd3, nd2);
g2.addEdge(ed1);
g2.addEdge(ed2);
g2.addEdge(ed3);
g2.addEdge(ed4);
g2.addEdge(ed5);
g2.removeNode("nd1");
ArrayList<Node> testNodeAL;
ArrayList<Edge> testEdgeAL;
testEdgeAL = new ArrayList<Edge>();
testEdgeAL.add(ed5);
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.1 Fail");
}
g2.addNode(nd3);
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.2 Fail");
}
g2.removeNode(nd2);
testEdgeAL = new ArrayList<Edge>();
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.3 Fail");
}
testNodeAL = new ArrayList<Node>();
testNodeAL.add(nd3);
if (!g2.getNodes().equals(testNodeAL)) {
System.out.println("Test2.4 Fail");
}
testEdgeAL = new ArrayList<Edge>();
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.5 Fail");
}
if (!g2.consistent()) {
System.out.println("Test2.5.0 Fail");
}
g2.clear();
Node dyNode1 = new Node("dyNode1");
Node dyNode2 = new Node("dyNode2");
g2.addNode(dyNode1);
g2.addNode(dyNode2);
Edge dyEdge1 = new Edge(dyNode1, dyNode2, "dyEdge2");
g2.addEdge(dyEdge1);
g2.removeEdge(dyEdge1);
if (dyEdge1.getFrom() != null) {
System.out.println("Test2.5.1 Fail");
}
if (dyEdge1.getTo() != null) {
System.out.println("Test2.5.2 Fail");
}
HashSet dyTest = new HashSet();
if (dyNode1.getEdgesFrom().equals(dyTest) == false) {
System.out.println("Test2.5.3 Fail");
}
if (dyNode1.getEdgesTo().equals(dyTest) == false) {
System.out.println("Test2.5.4 Fail");
}
if (dyNode2.getEdgesFrom().equals(dyTest) == false) {
System.out.println("Test2.5.5 Fail");
}
if (dyNode2.getEdgesTo().equals(dyTest) == false) {
System.out.println("Test2.5.6 Fail");
}
if (!g2.consistent()) {
System.out.println("Test2.5.7 Fail");
}
g2.clear();
g2.addNode(nd1);
g2.addNode(nd2);
g2.addNode(nd3);
ed1 = new Edge(nd1, nd2, "A", 1.1);
ed2 = new Edge(nd1, nd2, "B");
ed3 = new Edge(nd2, nd1, 1);
ed4 = new Edge(nd1, nd1);
ed5 = new Edge(nd3, nd2);
g2.addEdge(ed1);
g2.addEdge(ed2);
g2.addEdge(ed3);
g2.addEdge(ed4);
g2.addEdge(ed5);
// edge label and weight testing
if (!ed1.getLabel().equals("A")) {
System.out.println("Test2.5.1 Fail");
}
ed1.setLabel("C");
if (!ed1.getLabel().equals("C")) {
System.out.println("Test2.5.2 Fail");
}
if (!ed2.getLabel().equals("B")) {
System.out.println("Test2.5.3 Fail");
}
if (ed1.getWeight() != 1.1) {
System.out.println("Test2.5.4 Fail");
}
ed1.setWeight(2.2);
if (ed1.getWeight() != 2.2) {
System.out.println("Test2.5.5 Fail");
}
if (ed3.getWeight() != 1) {
System.out.println("Test2.5.6 Fail");
}
g2.removeEdge(ed2);
testEdgeAL = new ArrayList<Edge>();
testEdgeAL.add(ed1);
testEdgeAL.add(ed3);
testEdgeAL.add(ed4);
testEdgeAL.add(ed5);
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.6 Fail");
}
ArrayList<Node> testNodeAL2;
testNodeAL2 = new ArrayList<Node>();
testNodeAL2.add(nd1);
testNodeAL2.add(nd2);
testNodeAL2.add(nd3);
if (!g2.getNodes().equals(testNodeAL2)) {
System.out.println("Test2.7 Fail");
}
g2.removeEdge(ed4);
testEdgeAL.remove(ed4);
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.8 Fail");
}
if (!g2.getNodes().equals(testNodeAL2)) {
System.out.println("Test2.9 Fail");
}
ed4 = new Edge(nd3, nd3);
g2.addEdge(ed4);
testEdgeAL.add(ed4);
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.10 Fail");
}
if (!g2.getNodes().equals(testNodeAL2)) {
System.out.println("Test2.11 Fail");
}
if (!g2.consistent()) {
System.out.println("Test2.11.0 Fail");
}
g2.clear();
testEdgeAL = new ArrayList<Edge>();
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.12 Fail");
}
if (!g2.getNodes().equals(testEdgeAL)) {
System.out.println("Test2.13 Fail");
}
ed4 = new Edge(nd2, nd1);
g2.addNode(nd1);
g2.addNode(nd2);
g2.addNode(nd3);
g2.addEdge(ed4);
g2.removeEdge(ed4);
if (!g2.getEdges().equals(testEdgeAL)) {
System.out.println("Test2.14 Fail");
}
if (!g2.getNodes().equals(testNodeAL2)) {
System.out.println("Test2.15 Fail");
}
if (!g2.consistent()) {
System.out.println("Test2.15.0 Fail");
}
// score field testing
Graph gScore = new Graph();
Node ns1 = new Node("A");
Node ns2 = new Node();
Node ns3 = new Node("C");
Node ns4 = new Node("D");
gScore.addNode(ns1);
gScore.addNode(ns2);
gScore.addNode(ns3);
gScore.addNode(ns4);
Edge es1 = new Edge(ns1, ns2, "A", 1.1);
Edge es2 = new Edge(ns4, ns1);
Edge es3 = new Edge(ns1, ns3);
gScore.addEdge(es1);
gScore.addEdge(es2);
gScore.addEdge(es3);
if (ns1.getScore() != 0.0) {
System.out.println("Test2.16.1 Fail");
}
if (es1.getScore() != 0.0) {
System.out.println("Test2.16.2 Fail");
}
ns1.setScore(1.3);
es1.setScore(2.4);
if (ns1.getScore() != 1.3) {
System.out.println("Test2.16.3 Fail");
}
if (es1.getScore() != 2.4) {
System.out.println("Test2.16.4 Fail");
}
ArrayList<Node> alNodeScore = new ArrayList<Node>();
ArrayList<Edge> alEdgeScore = new ArrayList<Edge>();
gScore.setNodesScores(alNodeScore, 3.5);
gScore.setEdgesScores(alEdgeScore, 4.6);
if (ns1.getScore() != 1.3) {
System.out.println("Test2.16.5 Fail");
}
if (es1.getScore() != 2.4) {
System.out.println("Test2.16.6 Fail");
}
alNodeScore.add(ns1);
alNodeScore.add(ns2);
gScore.setNodesScores(alNodeScore, 5.7);
alNodeScore.remove(ns1);
alNodeScore.remove(ns2);
alEdgeScore.add(es1);
alEdgeScore.add(es2);
gScore.setEdgesScores(alEdgeScore, 6.8);
if (ns1.getScore() != 5.7) {
System.out.println("Test2.16.7 Fail");
}
if (ns3.getScore() != 0.0) {
System.out.println("Test2.16.8 Fail");
}
if (es1.getScore() != 6.8) {
System.out.println("Test2.16.9 Fail");
}
if (es3.getScore() != 0.0) {
System.out.println("Test2.16.10 Fail");
}
gScore.setNodesScores(7.9);
gScore.setEdgesScores(8.01);
if (ns1.getScore() != 7.9) {
System.out.println("Test2.16.11 Fail");
}
if (ns3.getScore() != 7.9) {
System.out.println("Test2.16.12 Fail");
}
if (es1.getScore() != 8.01) {
System.out.println("Test2.16.13 Fail");
}
if (es3.getScore() != 8.01) {
System.out.println("Test2.16.14 Fail");
}
Graph testChangeGraph = new Graph();
if (testChangeGraph.moveNodeToEnd(n1)) {
System.out.println("Test2.17.1 Fail");
}
if (testChangeGraph.moveEdgeToEnd(e1)) {
System.out.println("Test2.17.2 Fail");
}
ArrayList<Node> testChangeNodeAL = new ArrayList<Node>();
Node changeNode1 = new Node("c1");
testChangeGraph.addNode(changeNode1);
if (testChangeGraph.moveNodeToEnd(n1)) {
System.out.println("Test2.17.3 Fail");
}
testChangeGraph.moveNodeToEnd(changeNode1);
testChangeNodeAL = new ArrayList<Node>();
testChangeNodeAL.add(changeNode1);
if (!testChangeGraph.getNodes().equals(testChangeNodeAL)) {
System.out.println("Test2.17.4 Fail");
}
Node changeNode2 = new Node("c2");
Node changeNode3 = new Node("c3");
Edge changeEdge1 = new Edge(changeNode1, changeNode2);
Edge changeEdge2 = new Edge(changeNode3, changeNode2);
Edge changeEdge3 = new Edge(changeNode1, changeNode3);
testChangeGraph.addNode(changeNode2);
testChangeGraph.addNode(changeNode3);
testChangeGraph.addEdge(changeEdge1);
testChangeGraph.addEdge(changeEdge2);
testChangeGraph.addEdge(changeEdge3);
testChangeNodeAL = new ArrayList<Node>();
testChangeNodeAL.add(changeNode1);
testChangeNodeAL.add(changeNode3);
testChangeNodeAL.add(changeNode2);
testChangeGraph.moveNodeToEnd(changeNode2);
if (!testChangeGraph.getNodes().equals(testChangeNodeAL)) {
System.out.println("Test2.17.5 Fail");
}
ArrayList<Edge> testChangeEdgeAL = new ArrayList<Edge>();
testChangeEdgeAL.add(changeEdge2);
testChangeEdgeAL.add(changeEdge3);
testChangeEdgeAL.add(changeEdge1);
testChangeGraph.moveEdgeToEnd(changeEdge1);
if (!testChangeGraph.getEdges().equals(testChangeEdgeAL)) {
System.out.println("Test2.17.6 Fail");
}
System.out.print("Test 2 END");
System.out.println(" | Test 3 START: Shortest path");
ArrayList<Node> path = new ArrayList<Node>();
path.add(n5);
path.add(n4);
path.add(n1);
if (!g1.unweightedShortest(n5, n1).equals(path)) {
System.out.println("Test3.1 Fail");
}
path = new ArrayList<Node>();
path.add(n3);
if (!g1.unweightedShortest(n3, n3).equals(path)) {
System.out.println("Test3.2 Fail");
}
if (g1.unweightedShortest(n4, n6) != null) {
System.out.println("Test3.3 Fail");
}
System.out.print("Test 3 END");
// Adjacency Edge Graph creation stuff
System.out.println(" | Test 4 START: Adjacency Graph");
// finding or adding a node from a label
Node nA = g1.addAdjacencyNode("nA");
testNodeAL = new ArrayList<Node>();
testNodeAL.add(n1);
testNodeAL.add(n2);
testNodeAL.add(n3);
testNodeAL.add(n4);
testNodeAL.add(n5);
testNodeAL.add(n6);
testNodeAL.add(nA);
if (!g1.getNodes().equals(testNodeAL)) {
System.out.println("Test4.1 Fail");
}
if (g1.addAdjacencyNode("n2") != n2) {
System.out.println("Test4.2 Fail");
}
if (!g1.getNodes().equals(testNodeAL)) {
System.out.println("Test4.3 Fail");
}
Node nDuplicate = new Node("n1");
g1.addNode(nDuplicate);
if (g1.addAdjacencyNode("n1") != null) {
System.out.println("Test4.4 Fail");
}
g1.removeNode(nDuplicate);
if (!g1.consistent()) {
System.out.println("Test4.4.0 Fail");
}
// adding an adjacency edge
Edge eZ1 = g1.addAdjacencyEdge("n3", "n3");
Edge[] tempArray = {e1, e2, e3, e4, e5, e6, e7, e8, e9, eZ1};
ArrayList<Edge> testEdgeAL2 = new ArrayList<Edge>(Arrays.asList(tempArray));
if (!g1.getEdges().equals(testEdgeAL2)) {
System.out.println("Test4.5 Fail");
}
if (!g1.getNodes().equals(testNodeAL)) {
System.out.println("Test4.6 Fail");
}
g1.addAdjacencyEdge("nB", "");
if (g1.getNodes().size() != 8) {
System.out.println("Test4.7 Fail");
}
Edge eZ3 = g1.addAdjacencyEdge("nB", "n1");
Edge[] tempArray2 = {e1, e2, e3, e4, e5, e6, e7, e8, e9, eZ1, eZ3};
testEdgeAL2 = new ArrayList<Edge>(Arrays.asList(tempArray2));
if (!g1.getEdges().equals(testEdgeAL2)) {
System.out.println("Test4.8 Fail");
}
g1.addAdjacencyEdge("", "");
Edge eZ4 = g1.addAdjacencyEdge("nC", "nD");
Edge[] tempArray3 = {e1, e2, e3, e4, e5, e6, e7, e8, e9, eZ1, eZ3, eZ4};
testEdgeAL2 = new ArrayList<Edge>(Arrays.asList(tempArray3));
if (!g1.getEdges().equals(testEdgeAL2)) {
System.out.println("Test4.9 Fail");
}
if (g1.getNodes().size() != 10) {
System.out.println("Test4.10 Fail");
}
System.out.print("Test 4 END");
System.out.println(" | Test 5 START: Connected");
if (g1.connected()) {
System.out.println("Test5.1 Fail");
}
g1.addEdge(new Edge(nA, n6));
if (g1.connected()) {
System.out.println("Test5.2 Fail");
}
g1.addAdjacencyEdge("nA", "nD");
g1.addAdjacencyEdge("nA", "n4");
if (!g1.connected()) {
System.out.println("Test5.3 Fail");
}
if (!g1.consistent()) {
System.out.println("Test5.3.0 Fail");
}
Graph g4 = new Graph("g4");
if (!g4.connected()) {
System.out.println("Test5.4 Fail");
}
g4.addNode(new Node("z1"));
if (!g4.connected()) {
System.out.println("Test5.5 Fail");
}
g4.addNode(new Node("z2"));
if (g4.connected()) {
System.out.println("Test5.6 Fail");
}
g4.addAdjacencyEdge("z1", "z2");
if (!g4.connected()) {
System.out.println("Test5.7 Fail");
}
g4.addAdjacencyEdge("z2", "z3");
if (!g4.connected()) {
System.out.println("Test5.8 Fail");
}
g4.removeNode("z2");
if (g4.connected()) {
System.out.println("Test5.9 Fail");
}
System.out.print("Test 5 END");
System.out.println(" | Test 6 START: Equality by label testing");
Graph gc1 = new Graph("gc1");
Graph gc2 = new Graph("gc2");
if (!gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.1 Fail");
}
gc1.addNode(new Node("A"));
if (gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.2 Fail");
}
gc2.addNode(new Node("B"));
if (gc2.equalsByNodeLabel(gc1)) {
System.out.println("Test6.3 Fail");
}
gc2.addNode(new Node("A"));
if (gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.4 Fail");
}
gc1.addNode(new Node("B"));
if (!gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.5 Fail");
}
gc1.addNode(new Node("C"));
gc2.addNode(new Node("C"));
if (!gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.6 Fail");
}
gc1.addAdjacencyEdge("A", "B");
if (gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.7 Fail");
}
gc2.addAdjacencyEdge("A", "B");
if (!gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.8 Fail");
}
gc1.addAdjacencyEdge("C", "B");
gc2.addAdjacencyEdge("C", "B");
if (!gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.9 Fail");
}
gc1.addAdjacencyEdge("C", "A");
gc2.addAdjacencyEdge("A", "C");
if (gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.10 Fail");
}
// this test is two non isomorphic graphs for which the comparison succeeds.
gc1.clear();
gc2.clear();
Node gcn1 = new Node("");
Node gcn2 = new Node("");
Node gcn3 = new Node("");
Node gcn4 = new Node("");
gc1.addNode(gcn1);
gc1.addNode(gcn2);
gc2.addNode(gcn3);
gc2.addNode(gcn4);
gc1.addEdge(new Edge(gcn1, gcn2));
gc2.addEdge(new Edge(gcn3, gcn3));
if (!gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.11 Fail");
}
// dispite the equal labels these graphs are not equal due to the extra edge
gc2.addEdge(new Edge(gcn4, gcn3));
if (gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.12 Fail");
}
gc1.addEdge(new Edge(gcn1, gcn2));
if (!gc1.equalsByNodeLabel(gc2)) {
System.out.println("Test6.13 Fail");
}
if (!gc1.consistent()) {
System.out.println("Test6.13.0 Fail");
}
System.out.print("Test 6 END");
// Adjacency file testing. This test creates a file called test.adj
// It will crash with an exception if it cant create the file.
System.out.println(" | Test 7 START: Adjacency file - needs to read and write test.adj");
g4.generateRandomGraph(10, 15);
g4.saveAdjacencyFile("test.adj");
Graph g5 = new Graph("g5");
g5.loadAdjacencyFile("test.adj");
if (!g4.equalsByNodeLabel(g5)) {
System.out.println("Test7.1 Fail");
}
if (!g4.consistent()) {
System.out.println("Test7.1.0 Fail");
}
if (!g5.consistent()) {
System.out.println("Test7.1.1 Fail");
}
g4.clear();
g4.saveAdjacencyFile("test.adj");
g5.clear();
g5.generateRandomGraph(10, 15);
g4.loadAdjacencyFile("test.adj");
if (g4.equalsByNodeLabel(g5)) {
System.out.println("Test7.2 Fail");
}
g4.generateRandomGraph(20, 30);
g4.addNode(n1);
g4.saveAdjacencyFile("test.adj");
g5.generateRandomGraph(10, 15);
g5.loadAdjacencyFile("test.adj");
g4.loadAdjacencyFile("test.adj");
if (!g5.equalsByNodeLabel(g4)) {
System.out.println("Test7.3 Fail");
}
g4.clear();
g4.saveAdjacencyFile("test.adj");
g4.addNode(n1);
g4.loadAdjacencyFile("test.adj");
if (!g4.equalsByNodeLabel(new Graph())) {
System.out.println("Test7.4 Fail");
}
if (!g4.consistent()) {
System.out.println("Test7.4.0 Fail");
}
System.out.print("Test 7 END");
// partial node and edge access
System.out.println(" | Test 8 START: visited and path fields");
Graph g = new Graph();
HashSet<Node> testa = new HashSet<Node>();
HashSet<Node> testb = new HashSet<Node>();
if (!g.unvisitedNodes().equals(testa)) {
System.out.println("Test8.1 Fail");
}
if (!g.visitedNodes().equals(testb)) {
System.out.println("Test8.2 Fail");
}
g.addNode(n1);
g.addNode(n2);
g.addNode(n3);
g.addNode(n4);
Edge eV1 = new Edge(n1, n2);
Edge eV2 = new Edge(n3, n2);
Edge eV3 = new Edge(n4, n4);
testa.add(n1);
testa.add(n2);
testa.add(n3);
testa.add(n4);
if (!g.unvisitedNodes().equals(testa)) {
System.out.println("Test8.5 Fail");
}
if (!g.visitedNodes().equals(testb)) {
System.out.println("Test8.6 Fail");
}
n1.setVisited(true);
n2.setVisited(true);
testa.remove(n1);
testa.remove(n2);
testb.add(n1);
testb.add(n2);
if (!g.unvisitedNodes().equals(testa)) {
System.out.println("Test8.9 Fail");
}
if (!g.visitedNodes().equals(testb)) {
System.out.println("Test8.10 Fail");
}
HashSet<Edge> teste1 = new HashSet<Edge>();
HashSet<Edge> teste2 = new HashSet<Edge>();
if (!g.unvisitedEdges().equals(teste1)) {
System.out.println("Test8.13 Fail");
}
if (!g.visitedEdges().equals(teste2)) {
System.out.println("Test8.14 Fail");
}
g.addEdge(eV1);
g.addEdge(eV2);
g.addEdge(eV3);
eV1.setVisited(true);
eV2.setVisited(true);
eV3.setVisited(true);
teste2.add(eV1);
teste2.add(eV2);
teste2.add(eV3);
if (!g.unvisitedEdges().equals(teste1)) {
System.out.println("Test8.15 Fail");
}
if (!g.visitedEdges().equals(teste2)) {
System.out.println("Test8.16 Fail");
}
eV3.setVisited(false);
teste1.add(eV3);
teste2.remove(eV3);
if (!g.unvisitedEdges().equals(teste1)) {
System.out.println("Test8.17 Fail");
}
if (!g.visitedEdges().equals(teste2)) {
System.out.println("Test8.18 Fail");
}
System.out.print("Test 8 END");
// euler tour tests
System.out.println(" | Test 9 START: Euler Tour - needs to read and write test.euler");
Graph eg = new Graph();
ArrayList<Node> tour1 = new ArrayList<Node>();
eg.saveTour("test.euler", tour1);
ArrayList<Node> tour2 = eg.loadTour("test.euler");
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.1 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2.add(n1);
if (eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.2 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
eg.saveTour("test.euler", tour1, false);
tour2 = eg.loadTour("test.euler");
if (eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.3 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
eg = new Graph();
eg.generateRandomEulerGraph(5, 7);
tour1 = eg.euler();
eg.saveTour("test.euler", tour1);
tour2 = eg.loadTour("test.euler");
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.4 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(false);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.4a Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(true);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.4b Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2.remove(2);
if (eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.5 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
eg.addAdjacencyEdge("1", "2");
eg.saveTour("test.euler", tour1, false);
tour2 = eg.loadTour("test.euler");
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.6 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
if (!eg.consistent()) {
System.out.println("Test 9.6.0 Fail");
}
eg.clear();
eg.addNode(n1);
tour1 = eg.euler();
tour2 = new ArrayList<Node>();
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.6.1 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
eg.generateRandomEulerGraph(2, 4);
tour1 = eg.euler();
eg.saveTour("test.euler", tour1);
tour2 = eg.loadTour("test.euler");
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.7 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(false);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.7a Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(true);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.7b Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = new ArrayList<Node>();
if (eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.8 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
eg.addNode(new Node());
eg.saveTour("test.euler", tour1, false);
tour2 = eg.loadTour("test.euler");
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.9 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
if (!eg.consistent()) {
System.out.println("Test 9.9.0 Fail");
}
eg = new Graph();
eg.generateRandomEulerGraph(20, 30);
tour2 = eg.euler();
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.10 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(false);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.10a Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(true);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.10b Failed with tour " + tour2 + "\nand graph\n" + eg);
}
if (!eg.consistent()) {
System.out.println("Test 9.10.0 Fail");
}
Node nX0 = new Node("0");
Node nX1 = new Node("1");
Node nX2 = new Node("2");
Node nX3 = new Node("3");
Node nX4 = new Node("4");
eg = new Graph();
eg.addNode(nX0);
eg.addNode(nX1);
eg.addNode(nX2);
eg.addNode(nX3);
eg.addNode(nX4);
eg.addEdge(new Edge(nX0, nX1));
eg.addEdge(new Edge(nX3, nX2));
eg.addEdge(new Edge(nX4, nX2));
eg.addEdge(new Edge(nX0, nX1));
eg.addEdge(new Edge(nX3, nX0));
eg.addEdge(new Edge(nX0, nX3));
eg.addEdge(new Edge(nX3, nX4));
tour2 = eg.euler();
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.11 Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(false);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.11a Failed with tour " + tour2 + "\nand graph\n" + eg);
}
tour2 = eg.eulerSMK(true);
if (!eg.eulerTourInGraph(tour2)) {
System.out.println("Test 9.11b Failed with tour " + tour2 + "\nand graph\n" + eg);
}
System.out.print("Test 9 END");
// test the brute force tsp
System.out.println(" | Test 10 START: tsp");
Graph tspGraph = new Graph();
ArrayList<Edge> tspTest = new ArrayList<Edge>();
if (!tspGraph.tsp().equals(tspTest)) {
System.out.println("Test 10.1 Failed");
}
Node tspN1 = new Node("A");
tspGraph.addNode(tspN1);
if (!tspGraph.tsp().equals(tspTest)) {
System.out.println("Test 10.2 Failed");
}
Node tspN2 = new Node("B");
tspGraph.addNode(tspN2);
Edge tspE1 = new Edge(tspN1, tspN2, 2);
tspGraph.addEdge(tspE1);
tspTest.add(tspE1);
tspTest.add(tspE1);
if (!tspGraph.tsp().equals(tspTest)) {
System.out.println("Test 10.3 Failed");
}
if (tspGraph.sumEdgeWeights(tspGraph.tsp()) != 4.0) {
System.out.println("Test 10.4 Failed");
}
Node tspN3 = new Node("C");
tspGraph.addNode(tspN3);
Edge tspE2 = new Edge(tspN1, tspN3, 40);
Edge tspE3 = new Edge(tspN2, tspN3, 3);
tspGraph.addEdge(tspE2);
tspGraph.addEdge(tspE3);
if (tspGraph.sumEdgeWeights(tspGraph.tsp()) != 10.0) {
System.out.println("Test 10.5 Failed");
}
tspE2.setWeight(4);
if (tspGraph.sumEdgeWeights(tspGraph.tsp()) != 9.0) {
System.out.println("Test 10.6 Failed");
}
Node tspN4 = new Node("D");
tspGraph.addNode(tspN4);
tspGraph.addEdge(new Edge(tspN1, tspN4, 3.5));
tspGraph.addEdge(new Edge(tspN2, tspN4, 6));
tspGraph.addEdge(new Edge(tspN4, tspN3, 2.4));
if (tspGraph.sumEdgeWeights(tspGraph.tsp()) != 10.9) {
System.out.println("Test 10.7 Failed");
}
tspGraph.clear();
tspGraph.addAdjacencyEdge("A", "H", 17.0);
tspGraph.addAdjacencyEdge("G", "H", 1.0);
tspGraph.addAdjacencyEdge("H", "E", 2);
tspGraph.addAdjacencyEdge("G", "E", 9.0);
tspGraph.addAdjacencyEdge("E", "D", 20.0);
tspGraph.addAdjacencyEdge("C", "D", 2.5);
tspGraph.addAdjacencyEdge("C", "B", 3.0);
tspGraph.addAdjacencyEdge("D", "B", 4.0);
if (tspGraph.sumEdgeWeights(tspGraph.tsp()) != 89.5) {
System.out.println("Test 10.8 Failed");
}
tspGraph.clear();
tspGraph.addAdjacencyEdge("A", "H", "17", 17.0);
tspGraph.addAdjacencyEdge("G", "H", "1", 1.0);
tspGraph.addAdjacencyEdge("G", "A", "1", 1.0);
tspGraph.addAdjacencyEdge("E", "D", "20", 20.0);
tspGraph.addAdjacencyEdge("C", "D", "2 and a half", 2.5);
tspGraph.addAdjacencyEdge("C", "B", "3", 3.0);
tspGraph.addAdjacencyEdge("D", "B", "4", 4.0);
if (tspGraph.tsp() != null) {
System.out.println("Test 10.9 Failed");
}
System.out.print("Test 10 END");
System.out.println(" | Test 11 START: Union Find");
UnionFind uf = new UnionFind(10);
if (uf.find(9, 4)) {
System.out.println("Test 11.1 Fail");
}
uf.union(1, 2);
if (uf.getParent()[1] != 2) {
System.out.println("Test 11.2 Fail");
}
if (uf.getParent()[2] != uf.ROOT) {
System.out.println("Test 11.3 Fail");
}
uf.union(3, 2);
if (!uf.find(3, 2)) {
System.out.println("Test 11.4 Fail");
}
if (uf.find(3, 7)) {
System.out.println("Test 11.5 Fail");
}
if (uf.find(7, 3)) {
System.out.println("Test 11.6 Fail");
}
uf.union(5, 4);
if (!uf.find(5, 4)) {
System.out.println("Test 11.7 Fail");
}
if (uf.find(3, 4)) {
System.out.println("Test 11.8 Fail");
}
if (uf.find(5, 7)) {
System.out.println("Test 11.9 Fail");
}
uf.union(3, 4);
if (!uf.find(3, 4)) {
System.out.println("Test 11.10 Fail");
}
uf.union(4, 1);
if (!uf.find(4, 3)) {
System.out.println("Test 11.11 Fail");
}
uf.union(7, 6);
uf.union(8, 9);
uf.union(8, 7);
if (uf.find(7, 2)) {
System.out.println("Test 11.12 Fail");
}
uf.union(8, 4);
if (!uf.find(7, 2)) {
System.out.println("Test 11.13 Fail");
}
if (uf.find(7, 10)) {
System.out.println("Test 11.12 Fail");
}
System.out.print("Test 11 END");
System.out.println(" | Test 12 START: mst");
Graph mstGraph = new Graph();
// generating a complete graph
mstGraph.generateCompleteGraph(0);
if (mstGraph.getNodes().size() != 0) {
System.out.println("Test12.1.1 Fail");
}
if (mstGraph.getEdges().size() != 0) {
System.out.println("Test12.1.2 Fail");
}
mstGraph.generateCompleteGraph(1);
if (mstGraph.getNodes().size() != 1) {
System.out.println("Test12.1.3 Fail");
}
if (mstGraph.getEdges().size() != 0) {
System.out.println("Test12.1.4 Fail");
}
if (!mstGraph.consistent()) {
System.out.println("Test 12.1.4.0 Fail");
}
mstGraph.generateCompleteGraph(5);
if (mstGraph.getNodes().size() != 5) {
System.out.println("Test12.1.5 Fail");
}
if (mstGraph.getEdges().size() != 10) {
System.out.println("Test12.1.6 Fail");
}
mstGraph.generateCompleteGraph(8);
if (mstGraph.getNodes().size() != 8) {
System.out.println("Test12.1.7 Fail");
}
if (mstGraph.getEdges().size() != 28) {
System.out.println("Test12.1.8 Fail");
}
mstGraph.setEdgesWeights(1.0);
if (mstGraph.sumEdgeWeights(mstGraph.prim()) != 7) {
System.out.println("Test12.2.1 Fail");
}
if (mstGraph.sumEdgeWeights(mstGraph.kruskal()) != 7) {
System.out.println("Test12.2.2 Fail");
}
mstGraph.clear();
ArrayList<Edge> mstCompare = new ArrayList<Edge>();
if (!mstCompare.equals(mstGraph.prim())) {
System.out.println("Test12.2.3 Fail");
}
if (!mstCompare.equals(mstGraph.kruskal())) {
System.out.println("Test12.2.4 Fail");
}
Node mstN1 = new Node("A");
Node mstN2 = new Node("B");
Node mstN3 = new Node("C");
mstGraph.addNode(mstN1);
if (!mstCompare.equals(mstGraph.prim())) {
System.out.println("Test12.2.5 Fail");
}
if (!mstCompare.equals(mstGraph.kruskal())) {
System.out.println("Test12.2.6 Fail");
}
mstGraph.addNode(mstN2);
if (mstGraph.prim() != null) {
System.out.println("Test12.2.7 Fail");
}
if (mstGraph.kruskal() != null) {
System.out.println("Test12.2.8 Fail");
}
Edge mstE1 = new Edge(mstN1, mstN2, "A", 1);
mstGraph.addEdge(mstE1);
mstCompare.add(mstE1);
if (!mstCompare.equals(mstGraph.prim())) {
System.out.println("Test12.2.9 Fail");
}
if (!mstCompare.equals(mstGraph.kruskal())) {
System.out.println("Test12.2.10 Fail");
}
mstGraph.addNode(mstN3);
Edge mstE2 = new Edge(mstN2, mstN3, "B", 5);
Edge mstE3 = new Edge(mstN3, mstN1, "C", 2);
Edge mstE4 = new Edge(mstN2, mstN2, "D", 1);
mstGraph.addEdge(mstE2);
mstGraph.addEdge(mstE3);
mstGraph.addEdge(mstE4);
mstCompare.add(mstE3);
if (mstGraph.sumEdgeWeights(mstGraph.prim()) != 3) {
System.out.println("Test12.2.11 Fail");
}
if (mstGraph.sumEdgeWeights(mstGraph.kruskal()) != 3) {
System.out.println("Test12.2.12 Fail");
}
// generated graph test - randomly generate graphs of size
// 1-10 nodes, edges 4-14, and compare prim against kruskal
// on test fail, the graph should be printed for reference.
for (int i = 1; i <= 10; i++) {
mstGraph.generateRandomGraph(i, i + 3);
mstGraph.randomlyWeightGraph(0, 100);
ArrayList<Edge> prim = mstGraph.prim();
double primSum = 0;
if (prim != null) {
primSum = mstGraph.sumEdgeWeights(prim);
}
ArrayList<Edge> kruskal = mstGraph.kruskal();
double kruskalSum = 0;
if (kruskal != null) {
kruskalSum = mstGraph.sumEdgeWeights(kruskal);
}
if (primSum != kruskalSum) {
System.out.println("Test 12.3." + i + " Failed - mst sums not equal");
System.out.println("Randomly Generated Graph");
System.out.println(mstGraph);
}
if (!mstGraph.consistent()) {
System.out.println("Test 12.3." + i + ".0 Failed - graph consistency check failed");
System.out.println("Randomly Generated Graph");
System.out.println(mstGraph);
}
}
System.out.print("Test 12 END");
// test the types
System.out.println(" | Test 13 START: Node and Edge Types");
NodeType nt11 = new NodeType("nt11");
NodeType nt21 = new NodeType("nt21");
NodeType nt31 = new NodeType("nt31");
NodeType nt22 = new NodeType("nt22");
NodeType nt32 = new NodeType("nt32");
NodeType nt33 = new NodeType("nt33");
NodeType nt34 = new NodeType("nt34");
NodeType nt23 = new NodeType("nt23");
NodeType nt41 = new NodeType("nt41");
NodeType nt42 = new NodeType("nt42");
NodeType nt43 = new NodeType("nt43");
nt21.setParent(nt11);
nt31.setParent(nt21);
nt22.setParent(nt11);
nt32.setParent(nt21);
nt33.setParent(nt22);
nt34.setParent(nt21);
nt23.setParent(nt11);
nt41.setParent(nt32);
nt42.setParent(nt32);
nt43.setParent(nt32);
nt22.removeParent();
NodeType ntmove = (NodeType) nt33.getParent();
nt32.setParent(ntmove);
if (nt11.setParent(nt31)) {
System.out.println("Test 13.3.1 FAIL");
}
if (!nt11.setParent(nt43)) {
System.out.println("Test 13.3.2 FAIL");
}
if (nt43.setParent(nt43)) {
System.out.println("Test 13.3.3 FAIL");
}
if (nt43.setParent(nt43)) {
System.out.println("Test 13.3.4 FAIL");
}
if (!nt43.ancestor(nt32)) {
System.out.println("Test 13.3.5 FAIL");
}
if (nt41.ancestor(nt41)) {
System.out.println("Test 13.3.6 FAIL");
}
if (!nt33.ancestor(nt22)) {
System.out.println("Test 13.3.7 FAIL");
}
if (nt41.ancestor(nt42)) {
System.out.println("Test 13.3.8 FAIL");
}
if (nt41.root() != nt22) {
System.out.println("Test 13.3.9 FAIL");
}
if (nt22.root() != nt22) {
System.out.println("Test 13.3.10 FAIL");
}
EdgeType et11 = new EdgeType("et11");
EdgeType et21 = new EdgeType("et21");
EdgeType et22 = new EdgeType("et22");
EdgeType et31 = new EdgeType("et31");
et21.setParent(et11);
et31.setParent(et22);
et22.setParent(et11);
et21.setDirected(true);
if (!et31.ancestor(et11)) {
System.out.println("Test 13.4.1 FAIL");
}
if (et11.ancestor(et21)) {
System.out.println("Test 13.4.2 FAIL");
}
if (et21.root() != et11) {
System.out.println("Test 13.4.3 FAIL");
}
if (et11.root() != et11) {
System.out.println("Test 13.4.4 FAIL");
}
Node nt1 = new Node("nt1", new Point(100, 100));
Node nt2 = new Node("nt2", nt32, new Point(110, 100));
Node nt3 = new Node("nt3", nt11, new Point(200, 200));
Edge et1 = new Edge(nt1, nt2, "e1", 3.0, et11);
Edge et2 = new Edge(nt2, nt3, "e2");
Edge et3 = new Edge(nt1, nt3, "e3", 0.0, et31);
Graph gt = new Graph("gt1");
gt.addNode(nt1);
gt.addNode(nt2);
gt.addNode(nt3);
gt.addEdge(et1);
gt.addEdge(et2);
gt.addEdge(et3);
if (nt1.getType() != Graph.DEFAULT_NODE_TYPE) {
System.out.println("Test 13.5.1 FAIL");
}
if (nt2.getType() != nt32) {
System.out.println("Test 13.5.2 FAIL");
}
nt2.setType(nt41);
if (nt2.getType() != nt41) {
System.out.println("Test 13.5.3 FAIL");
}
if (et2.getType() != Graph.DEFAULT_EDGE_TYPE) {
System.out.println("Test 13.5.4 FAIL");
}
if (et3.getType() != et31) {
System.out.println("Test 13.5.5 FAIL");
}
et3.setType(Graph.DEFAULT_EDGE_TYPE);
if (et3.getType() != Graph.DEFAULT_EDGE_TYPE) {
System.out.println("Test 13.5.7 FAIL");
}
if (Graph.DEFAULT_EDGE_TYPE.root() != Graph.DEFAULT_EDGE_TYPE) {
System.out.println("Test 13.5.8 FAIL");
}
System.out.println("Test 13 END");
System.out.println("End Graph Tests");
}
////////////////////////////////////////////
// RFCI Algorithm 4.4 (Colombo et al, 2012)
// Orient colliders
////////////////////////////////////////////
private void ruleR0_RFCI(List<Node[]> rTuples) {
List<Node[]> lTuples = new ArrayList<Node[]>();
List<Node> nodes = graph.getNodes();
///////////////////////////////
// process tuples in rTuples
while (!rTuples.isEmpty()) {
Node[] thisTuple = rTuples.remove(0);
Node i = thisTuple[0];
Node j = thisTuple[1];
Node k = thisTuple[2];
final List<Node> nodes1 = getSepset(i, k);
if (nodes1 == null) continue;
List<Node> sepSet = new ArrayList<Node>(nodes1);
sepSet.remove(j);
boolean independent1 = false;
if (knowledge.noEdgeRequired(i.getName(), j.getName())) // if BK allows
{
try {
independent1 = independenceTest.isIndependent(i, j, sepSet);
} catch (Exception e) {
independent1 = true;
}
}
boolean independent2 = false;
if (knowledge.noEdgeRequired(j.getName(), k.getName())) // if BK allows
{
try {
independent2 = independenceTest.isIndependent(j, k, sepSet);
} catch (Exception e) {
independent2 = true;
}
}
if (!independent1 && !independent2) {
lTuples.add(thisTuple);
} else {
// set sepSets to minimal separating sets
if (independent1) {
setMinSepSet(sepSet, i, j);
graph.removeEdge(i, j);
}
if (independent2) {
setMinSepSet(sepSet, j, k);
graph.removeEdge(j, k);
}
// add new unshielded tuples to rTuples
for (Node thisNode : nodes) {
List<Node> adjacentNodes = graph.getAdjacentNodes(thisNode);
if (independent1) // <i, ., j>
{
if (adjacentNodes.contains(i) && adjacentNodes.contains(j)) {
Node[] newTuple = {i, thisNode, j};
rTuples.add(newTuple);
}
}
if (independent2) // <j, ., k>
{
if (adjacentNodes.contains(j) && adjacentNodes.contains(k)) {
Node[] newTuple = {j, thisNode, k};
rTuples.add(newTuple);
}
}
}
// remove tuples involving either (if independent1) <i, j>
// or (if independent2) <j, k> from rTuples
Iterator<Node[]> iter = rTuples.iterator();
while (iter.hasNext()) {
Node[] curTuple = iter.next();
if ((independent1 && (curTuple[1] == i) && ((curTuple[0] == j) || (curTuple[2] == j)))
|| (independent2 && (curTuple[1] == k) && ((curTuple[0] == j) || (curTuple[2] == j)))
|| (independent1 && (curTuple[1] == j) && ((curTuple[0] == i) || (curTuple[2] == i)))
|| (independent2
&& (curTuple[1] == j)
&& ((curTuple[0] == k) || (curTuple[2] == k)))) {
iter.remove();
}
}
// remove tuples involving either (if independent1) <i, j>
// or (if independent2) <j, k> from lTuples
iter = lTuples.iterator();
while (iter.hasNext()) {
Node[] curTuple = iter.next();
if ((independent1 && (curTuple[1] == i) && ((curTuple[0] == j) || (curTuple[2] == j)))
|| (independent2 && (curTuple[1] == k) && ((curTuple[0] == j) || (curTuple[2] == j)))
|| (independent1 && (curTuple[1] == j) && ((curTuple[0] == i) || (curTuple[2] == i)))
|| (independent2
&& (curTuple[1] == j)
&& ((curTuple[0] == k) || (curTuple[2] == k)))) {
iter.remove();
}
}
}
}
///////////////////////////////////////////////////////
// orient colliders (similar to original FCI ruleR0)
for (Node[] thisTuple : lTuples) {
Node i = thisTuple[0];
Node j = thisTuple[1];
Node k = thisTuple[2];
List<Node> sepset = getSepset(i, k);
if (sepset == null) {
continue;
}
if (!sepset.contains(j) && graph.isAdjacentTo(i, j) && graph.isAdjacentTo(j, k)) {
if (!isArrowpointAllowed(i, j)) {
continue;
}
if (!isArrowpointAllowed(k, j)) {
continue;
}
graph.setEndpoint(i, j, Endpoint.ARROW);
graph.setEndpoint(k, j, Endpoint.ARROW);
printWrongColliderMessage(i, j, k, "R0_RFCI");
}
}
}
public void testRemoveEdges() {
Graph graph = graphTest.getGraphInstance();
List<String> ids = generateIds(3);
Vertex v1 = graph.addVertex(convertId(ids.get(0)));
Vertex v2 = graph.addVertex(convertId(ids.get(1)));
Vertex v3 = graph.addVertex(convertId(ids.get(2)));
Edge e1 = graph.addEdge(null, v1, v2, convertId("knows"));
Edge e2 = graph.addEdge(null, v2, v3, convertId("pets"));
Edge e3 = graph.addEdge(null, v2, v3, convertId("cares_for"));
if (graphTest.supportsVertexIteration) assertEquals(3, count(graph.getVertices()));
graph.removeEdge(e1);
assertEquals(0, count(v1.getOutEdges()));
assertEquals(2, count(v2.getOutEdges()));
assertEquals(0, count(v3.getOutEdges()));
assertEquals(0, count(v1.getInEdges()));
assertEquals(0, count(v2.getInEdges()));
assertEquals(2, count(v3.getInEdges()));
if (!graphTest.ignoresSuppliedIds) {
v1 = graph.getVertex(convertId(ids.get(0)));
v2 = graph.getVertex(convertId(ids.get(1)));
v3 = graph.getVertex(convertId(ids.get(2)));
}
assertEquals(0, count(v1.getOutEdges()));
assertEquals(2, count(v2.getOutEdges()));
assertEquals(0, count(v3.getOutEdges()));
assertEquals(0, count(v1.getInEdges()));
assertEquals(0, count(v2.getInEdges()));
assertEquals(2, count(v3.getInEdges()));
graph.removeEdge(e2);
assertEquals(0, count(v1.getOutEdges()));
assertEquals(1, count(v2.getOutEdges()));
assertEquals(0, count(v3.getOutEdges()));
assertEquals(0, count(v1.getInEdges()));
assertEquals(0, count(v2.getInEdges()));
assertEquals(1, count(v3.getInEdges()));
if (!graphTest.ignoresSuppliedIds) {
v1 = graph.getVertex(convertId(ids.get(0)));
v2 = graph.getVertex(convertId(ids.get(1)));
v3 = graph.getVertex(convertId(ids.get(2)));
}
assertEquals(0, count(v1.getOutEdges()));
assertEquals(1, count(v2.getOutEdges()));
assertEquals(0, count(v3.getOutEdges()));
assertEquals(0, count(v1.getInEdges()));
assertEquals(0, count(v2.getInEdges()));
assertEquals(1, count(v3.getInEdges()));
graph.removeEdge(e3);
assertEquals(0, count(v1.getOutEdges()));
assertEquals(0, count(v2.getOutEdges()));
assertEquals(0, count(v3.getOutEdges()));
assertEquals(0, count(v1.getInEdges()));
assertEquals(0, count(v2.getInEdges()));
assertEquals(0, count(v3.getInEdges()));
if (!graphTest.ignoresSuppliedIds) {
v1 = graph.getVertex(convertId(ids.get(0)));
v2 = graph.getVertex(convertId(ids.get(1)));
v3 = graph.getVertex(convertId(ids.get(2)));
}
assertEquals(0, count(v1.getOutEdges()));
assertEquals(0, count(v2.getOutEdges()));
assertEquals(0, count(v3.getOutEdges()));
assertEquals(0, count(v1.getInEdges()));
assertEquals(0, count(v2.getInEdges()));
assertEquals(0, count(v3.getInEdges()));
graph.shutdown();
}
public void testRemoveEdges() {
Graph graph = graphTest.generateGraph();
Vertex v1 = graph.addVertex(convertId(graph, "1"));
Vertex v2 = graph.addVertex(convertId(graph, "2"));
Vertex v3 = graph.addVertex(convertId(graph, "3"));
Edge e1 = graph.addEdge(null, v1, v2, convertId(graph, "knows"));
Edge e2 = graph.addEdge(null, v2, v3, convertId(graph, "pets"));
Edge e3 = graph.addEdge(null, v2, v3, convertId(graph, "cares_for"));
if (graph.getFeatures().supportsVertexIteration) assertEquals(3, count(graph.getVertices()));
graph.removeEdge(e1);
assertEquals(0, count(v1.getEdges(Direction.OUT)));
assertEquals(2, count(v2.getEdges(Direction.OUT)));
assertEquals(0, count(v3.getEdges(Direction.OUT)));
assertEquals(0, count(v1.getEdges(Direction.IN)));
assertEquals(0, count(v2.getEdges(Direction.IN)));
assertEquals(2, count(v3.getEdges(Direction.IN)));
if (!graph.getFeatures().ignoresSuppliedIds) {
v1 = graph.getVertex(convertId(graph, "1"));
v2 = graph.getVertex(convertId(graph, "2"));
v3 = graph.getVertex(convertId(graph, "3"));
}
assertEquals(0, count(v1.getEdges(Direction.OUT)));
assertEquals(2, count(v2.getEdges(Direction.OUT)));
assertEquals(0, count(v3.getEdges(Direction.OUT)));
assertEquals(0, count(v1.getEdges(Direction.IN)));
assertEquals(0, count(v2.getEdges(Direction.IN)));
assertEquals(2, count(v3.getEdges(Direction.IN)));
graph.removeEdge(e2);
assertEquals(0, count(v1.getEdges(Direction.OUT)));
assertEquals(1, count(v2.getEdges(Direction.OUT)));
assertEquals(0, count(v3.getEdges(Direction.OUT)));
assertEquals(0, count(v1.getEdges(Direction.IN)));
assertEquals(0, count(v2.getEdges(Direction.IN)));
assertEquals(1, count(v3.getEdges(Direction.IN)));
if (!graph.getFeatures().ignoresSuppliedIds) {
v1 = graph.getVertex(convertId(graph, "1"));
v2 = graph.getVertex(convertId(graph, "2"));
v3 = graph.getVertex(convertId(graph, "3"));
}
assertEquals(0, count(v1.getEdges(Direction.OUT)));
assertEquals(1, count(v2.getEdges(Direction.OUT)));
assertEquals(0, count(v3.getEdges(Direction.OUT)));
assertEquals(0, count(v1.getEdges(Direction.IN)));
assertEquals(0, count(v2.getEdges(Direction.IN)));
assertEquals(1, count(v3.getEdges(Direction.IN)));
graph.removeEdge(e3);
assertEquals(0, count(v1.getEdges(Direction.OUT)));
assertEquals(0, count(v2.getEdges(Direction.OUT)));
assertEquals(0, count(v3.getEdges(Direction.OUT)));
assertEquals(0, count(v1.getEdges(Direction.IN)));
assertEquals(0, count(v2.getEdges(Direction.IN)));
assertEquals(0, count(v3.getEdges(Direction.IN)));
if (!graph.getFeatures().ignoresSuppliedIds) {
v1 = graph.getVertex(convertId(graph, "1"));
v2 = graph.getVertex(convertId(graph, "2"));
v3 = graph.getVertex(convertId(graph, "3"));
}
assertEquals(0, count(v1.getEdges(Direction.OUT)));
assertEquals(0, count(v2.getEdges(Direction.OUT)));
assertEquals(0, count(v3.getEdges(Direction.OUT)));
assertEquals(0, count(v1.getEdges(Direction.IN)));
assertEquals(0, count(v2.getEdges(Direction.IN)));
assertEquals(0, count(v3.getEdges(Direction.IN)));
graph.shutdown();
}
/** Do an actual deletion (Definition 13 from Chickering, 2002). */
private void delete(Node x, Node y, List<Node> subset, Graph graph, double bump) {
Edge trueEdge = null;
if (trueGraph != null) {
Node _x = trueGraph.getNode(x.getName());
Node _y = trueGraph.getNode(y.getName());
trueEdge = trueGraph.getEdge(_x, _y);
}
if (log && verbose) {
Edge oldEdge = graph.getEdge(x, y);
String label = trueGraph != null && trueEdge != null ? "*" : "";
TetradLogger.getInstance()
.log(
"deletedEdges",
(graph.getNumEdges() - 1)
+ ". DELETE "
+ oldEdge
+ " "
+ subset
+ " ("
+ bump
+ ") "
+ label);
out.println(
(graph.getNumEdges() - 1)
+ ". DELETE "
+ oldEdge
+ " "
+ subset
+ " ("
+ bump
+ ") "
+ label);
} else {
int numEdges = graph.getNumEdges() - 1;
if (numEdges % 50 == 0) out.println(numEdges);
}
graph.removeEdge(x, y);
for (Node h : subset) {
Edge oldEdge = graph.getEdge(y, h);
graph.removeEdge(y, h);
graph.addDirectedEdge(y, h);
if (log) {
TetradLogger.getInstance()
.log("directedEdges", "--- Directing " + oldEdge + " to " + graph.getEdge(y, h));
}
if (verbose) {
out.println("--- Directing " + oldEdge + " to " + graph.getEdge(y, h));
}
if (Edges.isUndirectedEdge(graph.getEdge(x, h))) {
if (!graph.isAdjacentTo(x, h))
throw new IllegalArgumentException("Not adjacent: " + x + ", " + h);
oldEdge = graph.getEdge(x, h);
graph.removeEdge(x, h);
graph.addDirectedEdge(x, h);
if (log) {
TetradLogger.getInstance()
.log("directedEdges", "--- Directing " + oldEdge + " to " + graph.getEdge(x, h));
}
if (verbose) {
out.println("--- Directing " + oldEdge + " to " + graph.getEdge(x, h));
}
}
}
}
// serial.
private void insert(Node x, Node y, List<Node> t, Graph graph, double bump) {
if (graph.isAdjacentTo(x, y)) {
return; // The initial graph may already have put this edge in the graph.
// throw new IllegalArgumentException(x + " and " + y + " are already adjacent in
// the graph.");
}
Edge trueEdge = null;
if (trueGraph != null) {
Node _x = trueGraph.getNode(x.getName());
Node _y = trueGraph.getNode(y.getName());
trueEdge = trueGraph.getEdge(_x, _y);
}
graph.addDirectedEdge(x, y);
if (log) {
String label = trueGraph != null && trueEdge != null ? "*" : "";
TetradLogger.getInstance()
.log(
"insertedEdges",
graph.getNumEdges()
+ ". INSERT "
+ graph.getEdge(x, y)
+ " "
+ t
+ " "
+ bump
+ " "
+ label);
} else {
int numEdges = graph.getNumEdges() - 1;
if (verbose) {
if (numEdges % 50 == 0) out.println(numEdges);
}
}
if (verbose) {
String label = trueGraph != null && trueEdge != null ? "*" : "";
out.println(
graph.getNumEdges()
+ ". INSERT "
+ graph.getEdge(x, y)
+ " "
+ t
+ " "
+ bump
+ " "
+ label);
} else {
int numEdges = graph.getNumEdges() - 1;
if (verbose) {
if (numEdges % 50 == 0) out.println(numEdges);
}
}
for (Node _t : t) {
Edge oldEdge = graph.getEdge(_t, y);
if (oldEdge == null) throw new IllegalArgumentException("Not adjacent: " + _t + ", " + y);
graph.removeEdge(_t, y);
graph.addDirectedEdge(_t, y);
if (log && verbose) {
TetradLogger.getInstance()
.log("directedEdges", "--- Directing " + oldEdge + " to " + graph.getEdge(_t, y));
out.println("--- Directing " + oldEdge + " to " + graph.getEdge(_t, y));
}
}
}
