コード例 #1
0
 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;
 }
コード例 #2
0
  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;
  }
コード例 #3
0
  @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));
  }