public Integer returnMetricaAresta(HashMap<ArestaVO, Integer> metrica, ArestaVO aresta) {
for (ArestaVO key : metrica.keySet()) {
if (key.getNo1().equals(aresta.getNo1()) && key.getNo2().equals(aresta.getNo2())) {
return metrica.get(key);
}
}
return -1;
}
private Grafo populaGrafo() {
Grafo grafo = new Grafo();
ArrayList<No> noList = new ArrayList<No>();
for (int i = 0; i < 7; i++) {
No no = new No();
no.setNome(String.valueOf(i));
noList.add(no);
}
ArrayList<ArestaVO> arestaList = new ArrayList<ArestaVO>();
ArestaVO aresta = new ArestaVO();
aresta.setNo1("0");
aresta.setNo2("2");
aresta.setPeso(5);
arestaList.add(aresta);
ArestaVO aresta1 = new ArestaVO();
aresta1.setNo1("0");
aresta1.setNo2("3");
aresta1.setPeso(8);
arestaList.add(aresta1);
ArestaVO aresta2 = new ArestaVO();
aresta2.setNo1("2");
aresta2.setNo2("3");
aresta2.setPeso(10);
arestaList.add(aresta2);
ArestaVO aresta3 = new ArestaVO();
aresta3.setNo1("2");
aresta3.setNo2("4");
aresta3.setPeso(3);
arestaList.add(aresta3);
ArestaVO aresta4 = new ArestaVO();
aresta4.setNo1("2");
aresta4.setNo2("1");
aresta4.setPeso(16);
arestaList.add(aresta4);
ArestaVO aresta5 = new ArestaVO();
aresta5.setNo1("3");
aresta5.setNo2("4");
aresta5.setPeso(2);
arestaList.add(aresta5);
ArestaVO aresta6 = new ArestaVO();
aresta6.setNo1("3");
aresta6.setNo2("5");
aresta6.setPeso(18);
arestaList.add(aresta6);
ArestaVO aresta7 = new ArestaVO();
aresta7.setNo1("4");
aresta7.setNo2("1");
aresta7.setPeso(30);
arestaList.add(aresta7);
ArestaVO aresta8 = new ArestaVO();
aresta8.setNo1("4");
aresta8.setNo2("6");
aresta8.setPeso(14);
arestaList.add(aresta8);
ArestaVO aresta9 = new ArestaVO();
aresta9.setNo1("4");
aresta9.setNo2("5");
aresta9.setPeso(12);
arestaList.add(aresta9);
ArestaVO aresta10 = new ArestaVO();
aresta10.setNo1("1");
aresta10.setNo2("6");
aresta10.setPeso(26);
arestaList.add(aresta10);
ArestaVO aresta11 = new ArestaVO();
aresta11.setNo1("5");
aresta11.setNo2("6");
aresta11.setPeso(4);
arestaList.add(aresta11);
grafo.setNos(noList);
grafo.setArestas(arestaList);
return grafo;
}
@Test
public void simpleTest() {
Grafo grafo = populaGrafo();
ArrayList<ArestaVO> arestaAGM = agm.generate(grafo);
HashMap<ArestaVO, Integer> metrica =
edgeCentralityMetricaService.calcula(arestaAGM, geraNomeBairroMap());
ArestaVO aresta = new ArestaVO();
aresta.setNo1("0");
aresta.setNo2("3");
Assert.assertEquals(new Integer("6"), returnMetricaAresta(metrica, aresta));
aresta.setNo1("3");
aresta.setNo2("5");
Assert.assertEquals(new Integer("10"), returnMetricaAresta(metrica, aresta));
aresta.setNo1("4");
aresta.setNo2("5");
Assert.assertEquals(new Integer("12"), returnMetricaAresta(metrica, aresta));
aresta.setNo1("4");
aresta.setNo2("1");
Assert.assertEquals(new Integer("12"), returnMetricaAresta(metrica, aresta));
aresta.setNo1("2");
aresta.setNo2("1");
Assert.assertEquals(new Integer("6"), returnMetricaAresta(metrica, aresta));
aresta.setNo1("1");
aresta.setNo2("6");
Assert.assertEquals(new Integer("6"), returnMetricaAresta(metrica, aresta));
}