Method testPi

test/org/apache/pig/test/TestPi.java:91–125

()

    }

    @Test
    public void testPi() throws Exception {

        pig.registerQuery("A = load 'foo' using mock.Storage() as (x:double, y:double);");
        pig.registerQuery("B = foreach A generate $0 - 0.5 as d1, $1 - 0.5 as d2;");
        pig.registerQuery("C = foreach B generate $0 * $0 as m1, $1 * $1 as m2;");
        pig.registerQuery("D = foreach C generate $0 + $1 as s1;");
        pig.registerQuery("D = foreach D generate $0, ARITY($0);");

        pig.registerQuery("E = filter D by $0 <= 0.25;");

        pig.registerQuery("F = cogroup D by $1, E by $1;");
        pig.registerQuery("G = foreach F generate COUNT($1) as total, COUNT($2) as in_circle;");

        Iterator<Tuple> values = pig.openIterator("G");
        assertTrue(values.hasNext());
        Tuple finalValues = values.next();
        int totalPoints = DataType.toInteger(finalValues.get(0));
        int inCirclePoints = DataType.toInteger(finalValues.get(1));

        log.info("Value of PI = " + 4 * (double)inCircle / (double)total);
        log.info("Value of PI (From Test data) = " + 4 * (double)inCirclePoints / (double)totalPoints);


        Iterator<Tuple> lengthTest = pig.openIterator("D");

        while(lengthTest.hasNext()) {
            Tuple temp = lengthTest.next();
            totalLengthTest += temp.get(0).toString().length();
        }

        assertEquals(totalPoints, total);
        assertEquals(inCirclePoints, inCircle);
        assertEquals(totalLengthTest, totalLength);
    }
}