Fixed yacc tests to account for dict hash key randomization

1 files changed, 21 insertions, 3 deletions

diff --git a/test/testyacc.py b/test/testyacc.py
index 1a98b4f..4329d3b 100644
--- a/test/testyacc.py
+++ b/test/testyacc.py
@@ -34,7 +34,7 @@ def pymodule_out_exists(filename):
 def pymodule_out_remove(filename):
     os.remove(make_pymodule_path(filename))
 
-
+# Old implementation (not safe for Python 3.3)
 def check_expected(result,expected):
     resultlines = []
     for line in result.splitlines():
@@ -52,6 +52,26 @@ def check_expected(result,expected):
             return False
     return True
 
+# Check the output to see if it contains all of a set of expected output lines.
+# This alternate implementation looks weird, but is needed to properly handle
+# some variations in error message order that occurs due to dict hash table
+# randomization that was introduced in Python 3.3
+def check_expected(result, expected):
+    resultlines = set()
+    for line in result.splitlines():
+        if line.startswith("WARNING: "):
+            line = line[9:]
+        elif line.startswith("ERROR: "):
+            line = line[7:]
+        resultlines.add(line)
+
+    # Selectively remove expected lines from the output
+    for eline in expected.splitlines():
+        resultlines = set(line for line in resultlines if not line.endswith(eline))
+
+    # Return True if no result lines remain
+    return not bool(resultlines)
+
 def run_import(module):
     code = "import "+module
     exec(code)
@@ -371,6 +391,4 @@ class YaccErrorWarningTests(unittest.TestCase):
                                     "Precedence rule 'left' defined for unknown symbol '/'\n"
                                     ))
 
-
-            
 unittest.main()