Wenzel Jakob | 6d6fd09 | 2015-10-01 17:34:26 +0200 | [diff] [blame] | 1 | import sys |
| 2 | import os |
| 3 | import re |
| 4 | import subprocess |
| 5 | |
| 6 | remove_unicode_marker = re.compile(r'u(\'[^\']*\')') |
| 7 | remove_long_marker = re.compile(r'([0-9])L') |
Wenzel Jakob | 607654f | 2015-10-13 23:58:10 +0200 | [diff] [blame] | 8 | remove_hex = re.compile(r'0x[0-9a-fA-F]+') |
Wenzel Jakob | 6d6fd09 | 2015-10-01 17:34:26 +0200 | [diff] [blame] | 9 | shorten_floats = re.compile(r'([1-9][0-9]*\.[0-9]{4})[0-9]*') |
| 10 | |
| 11 | |
| 12 | def sanitize(lines): |
| 13 | lines = lines.split('\n') |
| 14 | for i in range(len(lines)): |
| 15 | line = lines[i] |
| 16 | if line.startswith(" |"): |
| 17 | line = "" |
| 18 | line = remove_unicode_marker.sub(r'\1', line) |
| 19 | line = remove_long_marker.sub(r'\1', line) |
Wenzel Jakob | 3b806d4 | 2015-10-11 16:29:35 +0200 | [diff] [blame] | 20 | line = remove_hex.sub(r'0', line) |
Wenzel Jakob | 6d6fd09 | 2015-10-01 17:34:26 +0200 | [diff] [blame] | 21 | line = shorten_floats.sub(r'\1', line) |
| 22 | line = line.replace('__builtin__', 'builtins') |
| 23 | line = line.replace('example.', '') |
| 24 | line = line.replace('method of builtins.PyCapsule instance', '') |
| 25 | line = line.strip() |
Wenzel Jakob | 607654f | 2015-10-13 23:58:10 +0200 | [diff] [blame] | 26 | if sys.platform == 'win32': |
| 27 | lower = line.lower() |
Wenzel Jakob | fab881c | 2015-10-18 17:04:24 +0200 | [diff] [blame] | 28 | if 'constructor' in lower or 'destructor' in lower \ |
| 29 | or 'ref' in lower: |
Wenzel Jakob | 607654f | 2015-10-13 23:58:10 +0200 | [diff] [blame] | 30 | line = "" |
Wenzel Jakob | 6d6fd09 | 2015-10-01 17:34:26 +0200 | [diff] [blame] | 31 | lines[i] = line |
| 32 | |
| 33 | lines = '\n'.join(sorted([l for l in lines if l != ""])) |
| 34 | |
| 35 | print('==================') |
| 36 | print(lines) |
| 37 | return lines |
Wenzel Jakob | 38bd711 | 2015-07-05 20:05:44 +0200 | [diff] [blame] | 38 | |
| 39 | path = os.path.dirname(__file__) |
| 40 | if path != '': |
| 41 | os.chdir(path) |
| 42 | |
| 43 | name = sys.argv[1] |
Wenzel Jakob | fab881c | 2015-10-18 17:04:24 +0200 | [diff] [blame] | 44 | output_bytes = subprocess.check_output([sys.executable, name + ".py"], |
| 45 | stderr=subprocess.STDOUT) |
| 46 | |
Wenzel Jakob | 6d6fd09 | 2015-10-01 17:34:26 +0200 | [diff] [blame] | 47 | output = sanitize(output_bytes.decode('utf-8')) |
| 48 | reference = sanitize(open(name + '.ref', 'r').read()) |
Wenzel Jakob | 38bd711 | 2015-07-05 20:05:44 +0200 | [diff] [blame] | 49 | |
Wenzel Jakob | fab881c | 2015-10-18 17:04:24 +0200 | [diff] [blame] | 50 | if 'NumPy missing' in output: |
| 51 | print('Test "%s" could not be run.' % name) |
| 52 | exit(0) |
| 53 | elif output == reference: |
Wenzel Jakob | 38bd711 | 2015-07-05 20:05:44 +0200 | [diff] [blame] | 54 | print('Test "%s" succeeded.' % name) |
| 55 | exit(0) |
| 56 | else: |
| 57 | print('Test "%s" FAILED!' % name) |
| 58 | exit(-1) |