PIG-5060: TestPigRunner.testDisablePigCounters2 failing with tez (knoguchi)


git-svn-id: https://svn.apache.org/repos/asf/pig/trunk@1770018 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/CHANGES.txt b/CHANGES.txt
index 8dd2a6f..22c95c8 100644
--- a/CHANGES.txt
+++ b/CHANGES.txt
@@ -59,6 +59,8 @@
  
 BUG FIXES
 
+PIG-5060: TestPigRunner.testDisablePigCounters2 failing with tez (knoguchi)
+
 PIG-5056: Fix AvroStorage writing enums (szita via daijy)
 
 PIG-5055: Infinite loop with join by fixed index (knoguchi)
diff --git a/test/org/apache/pig/test/TestPigRunner.java b/test/org/apache/pig/test/TestPigRunner.java
index 00b78b6..13ed468 100644
--- a/test/org/apache/pig/test/TestPigRunner.java
+++ b/test/org/apache/pig/test/TestPigRunner.java
@@ -1055,8 +1055,13 @@
             List<OutputStats> outputs = stats.getOutputStats();
             assertEquals(2, outputs.size());
             if (execType.equals("tez")) {
-                assertEquals(outputs.get(0).getNumberRecords(), 5);
-                assertEquals(outputs.get(1).getNumberRecords(), 2);
+                if( outputs.get(0).getLocation().endsWith("tmp/output") ) {
+                    assertEquals(2, outputs.get(0).getNumberRecords());
+                    assertEquals(5, outputs.get(1).getNumberRecords());
+                } else {
+                    assertEquals(5, outputs.get(0).getNumberRecords());
+                    assertEquals(2, outputs.get(1).getNumberRecords());
+                }
             } else {
                 for (OutputStats outstats : outputs) {
                     // the multi-output counters are disabled