Ejemplo n.º 1
0
  @Test
  public void testFilter34() throws Exception {
    System.out.println("testFilter34");

    PigServer pigServer = new PigServer(pigContext);

    pigServer.registerQuery(
        "G1 = load "
            + GalaxyPair3
            + " using PigStorage() as (objID1 : long, ra : float, dec : float, modelMag_u1 : float, modelMag_g1 : float, modelMag_r1 : float, modelMag_i1 : float, modelMag_z1 : float, petroR50_r1 : float, petrorad_u1 : float, petrorad_g1 : float, petrorad_r1 : float, petrorad_i1 : float, petrorad_z1 : float, petroRadErr_g1 : float, petroMag_g1 : float);");
    pigServer.registerQuery(
        "AG1 = filter G1 by modelMag_g1 > -9999.0;"); // petrorad_u1 > 0.0 and petrorad_g1 > 0.0 and
                                                      // petrorad_r1 > 0.0 and petrorad_i1 > 0.0 and
                                                      //  petrorad_z1 > 0.0 and petroRadErr_g1 > 0.0
                                                      // and petroMag_g1 >= 16.0 and  petroMag_g1 <=
                                                      // 21.0 and modelMag_u1 > -9999.0 and
                                                      // modelMag_g1 > -9999.0 and modelMag_r1 >
                                                      // -9999.0 and modelMag_i1 > -9999.0 and
                                                      // modelMag_z1 > -9999.0
    pigServer.registerQuery(
        "G2 = load "
            + GalaxyPair4
            + " using PigStorage() as (objID2 : long, ra : float, dec : float, modelMag_u2 : float, modelMag_g2 : float, modelMag_r2 : float, modelMag_i2 : float, modelMag_z2 : float, petroR50_r2 : float, petrorad_u2 : float, petrorad_g2 : float, petrorad_r2 : float, petrorad_i2 : float, petrorad_z2 : float, petroRadErr_g2: float, petroMag_g2 : float);");
    pigServer.registerQuery(
        "AG2 = filter G2 by modelMag_g2 > -9999.0;"); // and petrorad_u2 > 0.0 and petrorad_g2 > 0.0
                                                      // and petrorad_r2 > 0.0 and petrorad_i2 > 0.0
                                                      // and  petrorad_z2 > 0.0 and  petroRadErr_g2
                                                      // > 0.0 and petroMag_g2 >= 16.0 and
                                                      // petroMag_g2 <= 21.0 and modelMag_u2 >
                                                      // -9999.0 and modelMag_r2 > -9999.0 and
                                                      // modelMag_i2 > -9999.0 and modelMag_z2 >
                                                      // -9999.0
    pigServer.registerQuery(
        "N = load "
            + neighbors
            + " using PigStorage() as (objID3 : long, NeighborObjID : long, neighborType : int, distance : float);");
    pigServer.registerQuery("AN = filter N by neighborType == 3;");
    pigServer.registerQuery("D = join AN by objID3, AG1 by objID1;");
    pigServer.registerQuery("E = join D by NeighborObjID, AG2 by objID2;");
    pigServer.registerQuery("F = filter E by objID1 < objID2;"); //
    pigServer.registerQuery(
        "H = filter F by modelMag_g1 - modelMag_g2 > 3.0 or modelMag_g1 - modelMag_g2 < -3.0;");
    pigServer.registerQuery(
        "I = filter H by petroR50_r1 >= 0.25*petroR50_r2 and petroR50_r1 <= 4.0*petroR50_r2 and petroR50_r2 >= 0.25*petroR50_r1;");

    Map<Operator, DataBag> derivedData = pigServer.getExamples("I");

    assertTrue(derivedData != null);
  } //