query.append("A = LOAD 'foo' USING mock.Storage() AS (i:int);\n");
query.append("ASSERT A BY i > 0;\n");
query.append("STORE A INTO 'bar' USING mock.Storage();");
InputStream is = new ByteArrayInputStream(query.toString().getBytes());
pigServer.registerScript(is);
List<Tuple> out = data.get("bar");
assertEquals(3, out.size());
assertEquals(tuple(1), out.get(0));
assertEquals(tuple(2), out.get(1));