Search for test files in sys.path

1 files changed, 10 insertions, 0 deletions

diff --git a/Lib/test/test_rgbimg.py b/Lib/test/test_rgbimg.py
index 7418e69868..cdadc660dd 100644
--- a/Lib/test/test_rgbimg.py
+++ b/Lib/test/test_rgbimg.py
@@ -6,7 +6,17 @@ error = 'test_rgbimg.error'
 
 print 'RGBimg test suite:'
 
+def findfile(file):
+	if os.path.isabs(file): return file
+	import sys
+	for dn in sys.path:
+		fn = os.path.join(dn, file)
+		if os.path.exists(fn): return fn
+	return file
+
 def testimg(rgb_file, raw_file):
+	rgb_file = findfile(rgb_file)
+	raw_file = findfile(raw_file)
 	width, height = rgbimg.sizeofimage(rgb_file)
 	rgb = rgbimg.longimagedata(rgb_file)
 	if len(rgb) != width * height * 4: