matplotlib/tools/visualize_tests.py at master · SMSinclair/matplotlib · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

#!/usr/bin/env python

# This builds a html page of all images from the image comparison tests

# and opens that page in the browser.

# $ python tools/visualize_tests.py

import argparse

import os

from collections import defaultdict

html_template = """<html><head><style media="screen" type="text/css">

img{{

width:100%;

max-width:800px;

}}

</style>

</head><body>

{failed}

{body}

</body></html>

"""

subdir_template = """<h2>{subdir}</h2><table>

<thead><td>name</td><td>actual</td><td>expected</td><td>diff</td></thead>

{rows}

</table>

"""

failed_template = """<h2>Only Failed</h2><table>

<thead><td>name</td><td>actual</td><td>expected</td><td>diff</td></thead>

{rows}

</table>

"""

row_template = ('<tr>'

'<td>{0}{1}</td>'

'<td>{2}</td>'

'<td><a href="{3}"><img src="{3}"></a></td>'

'<td>{4}</td>'

'</tr>')

linked_image_template = '<a href="{0}"><img src="{0}"></a>'

def run(show_browser=True):

"""

Build a website for visual comparison

"""

image_dir = "result_images"

_subdirs = (name

for name in os.listdir(image_dir)

if os.path.isdir(os.path.join(image_dir, name)))

failed_rows = []

body_sections = []

for subdir in sorted(_subdirs):

if subdir == "test_compare_images":

# These are the images which test the image comparison functions.

continue

pictures = defaultdict(dict)

for file in os.listdir(os.path.join(image_dir, subdir)):

if os.path.isdir(os.path.join(image_dir, subdir, file)):

continue

fn, fext = os.path.splitext(file)

if fext != ".png":

continue

# Always use / for URLs.

if "-failed-diff" in fn:

pictures[fn[:-12]]["f"] = "/".join((subdir, file))

elif "-expected" in fn:

pictures[fn[:-9]]["e"] = "/".join((subdir, file))

else:

pictures[fn]["c"] = "/".join((subdir, file))

subdir_rows = []

for name, test in sorted(pictures.items()):

expected_image = test.get('e', '')

actual_image = test.get('c', '')

if 'f' in test:

# A real failure in the image generation, resulting in

# different images.

status = " (failed)"

failed = '<a href="{0}">diff</a>'.format(test['f'])

current = linked_image_template.format(actual_image)

failed_rows.append(row_template.format(name, "", current,

expected_image, failed))

elif 'c' not in test:

# A failure in the test, resulting in no current image

status = " (failed)"

failed = '--'

current = '(Failure in test, no image produced)'

failed_rows.append(row_template.format(name, "", current,

expected_image, failed))

else:

status = " (passed)"

failed = '--'

current = linked_image_template.format(actual_image)

subdir_rows.append(row_template.format(name, status, current,

expected_image, failed))

body_sections.append(

subdir_template.format(subdir=subdir, rows='\n'.join(subdir_rows)))

if failed_rows:

failed = failed_template.format(rows='\n'.join(failed_rows))

else:

failed = ''

body = ''.join(body_sections)

html = html_template.format(failed=failed, body=body)

index = os.path.join(image_dir, "index.html")

with open(index, "w") as f:

f.write(html)

show_message = not show_browser

if show_browser:

try:

import webbrowser

webbrowser.open(index)

except Exception:

show_message = True

if show_message:

print("Open {} in a browser for a visual comparison.".format(index))

if __name__ == '__main__':

parser = argparse.ArgumentParser()

parser.add_argument('--no-browser', action='store_true',

help="Don't show browser after creating index page.")

args = parser.parse_args()

run(show_browser=not args.no_browser)

matplotlib/tools/visualize_tests.py at master · SMSinclair/matplotlib · GitHub

Repository navigation

Expand file tree

Breadcrumbs

visualize_tests.py

Latest commit

History

Breadcrumbs

visualize_tests.py

File metadata and controls