diff --git a/test/testpy/__init__.py b/test/testpy/__init__.py
index 7ba9674d7d6e57..7d23c47214eedb 100644
--- a/test/testpy/__init__.py
+++ b/test/testpy/__init__.py
@@ -31,6 +31,11 @@
 import re
 import ast
 
+try:
+  reduce
+except NameError:
+  from functools import reduce
+
 
 FLAGS_PATTERN = re.compile(r"//\s+Flags:(.*)")
 FILES_PATTERN = re.compile(r"//\s+Files:(.*)")