* @throws Exception
*/
@Test
public void testWithSplit() throws Exception {
PigServer pigServer = new PigServer(ExecType.LOCAL);
Data data = resetData(pigServer);
data.set("foo",
tuple(1),
tuple(2),
tuple(3),
tuple(4),
tuple(5)
);
pigServer.registerQuery("A = LOAD 'foo' USING mock.Storage() AS (i:int);");
pigServer.registerQuery("SPLIT A INTO B IF i IN (1, 3, 5), C OTHERWISE;");
pigServer.registerQuery("STORE B INTO 'odd' USING mock.Storage();");
pigServer.registerQuery("STORE C INTO 'even' USING mock.Storage();");
List<Tuple> out = data.get("odd");
assertEquals(3, out.size());
assertEquals(tuple(1), out.get(0));
assertEquals(tuple(3), out.get(1));
assertEquals(tuple(5), out.get(2));
out = data.get("even");
assertEquals(2, out.size());
assertEquals(tuple(2), out.get(0));
assertEquals(tuple(4), out.get(1));
}