assertPlanContains(tezPlanContainer.getRoots().get(0).getTezOperPlan(), expectedFiles, size);
}
private void assertPlanContains(TezOperPlan plan, String[] expectedFiles, int size) throws VisitorException {
TezPOUserFuncVisitor udfVisitor = new TezPOUserFuncVisitor(plan);
udfVisitor.visit();
List<String> shipFiles = new ArrayList<String>();
shipFiles.addAll(udfVisitor.getShipFiles());
Assert.assertEquals(shipFiles.size(), size);