Source code

Revision control

Copy as Markdown

Other Tools

"""Generates Canvas tests from YAML file definitions."""
# Current code status:
# This was originally written by Philip Taylor for use at
# It has been adapted for use with the Web Platform Test Suite suite at
# The original version had a number of now-removed features (multiple versions
# of each test case of varying verbosity, Mozilla mochitests, semi-automated
# test harness). It also had a different directory structure.
# To update or add test cases:
# * Modify the tests*.yaml files.
# - 'name' is an arbitrary hierarchical name to help categorise tests.
# - 'desc' is a rough description of what behaviour the test aims to test.
# - 'code' is JavaScript code to execute, with some special commands starting
# with '@'.
# - 'expected' is what the final canvas output should be: a string 'green' or
# 'clear' (100x50 images in both cases), or a string 'size 100 50' (or any
# other size) followed by Python code using Pycairo to generate the image.
# * Run "./".
# This requires a few Python modules which might not be ubiquitous.
# It will usually emit some warnings, which ideally should be fixed but can
# generally be safely ignored.
# * Test the tests, add new ones to Git, remove deleted ones from Git, etc.
from typing import Any, DefaultDict, FrozenSet, List, Mapping, MutableMapping
from typing import Optional, Set, Tuple
import re
import collections
import copy
import dataclasses
import enum
import importlib
import itertools
import os
import pathlib
import sys
import textwrap
import jinja2
import cairocffi as cairo # type: ignore
except ImportError:
import cairo
# Compatible and lots faster.
import syck as yaml # type: ignore
except ImportError:
import yaml
class Error(Exception):
"""Base class for all exceptions raised by this module"""
class InvalidTestDefinitionError(Error):
"""Raised on invalid test definition."""
def _double_quote_escape(string: str) -> str:
return string.replace('\\', '\\\\').replace('"', '\\"')
def _escape_js(string: str) -> str:
string = _double_quote_escape(string)
# Kind of an ugly hack, for nicer failure-message output.
string = re.sub(r'\[(\w+)\]', r'[\\""+(\1)+"\\"]', string)
return string
def _unroll(text: str) -> str:
"""Unrolls text with all possible permutations of the parameter lists.
>>> print _unroll('f = {<a | b>: <1 | 2 | 3>};')
// a
f = {a: 1};
f = {a: 2};
f = {a: 3};
// b
f = {b: 1};
f = {b: 2};
f = {b: 3};
patterns = [] # type: List[Tuple[str, List[str]]]
while True:
match ='<([^>]+)>', text)
if not match:
key = f'@unroll_pattern_{len(patterns)}'
values = text[match.start(1):match.end(1)]
text = text[:match.start(0)] + key + text[match.end(0):]
patterns.append((key, [value.strip() for value in values.split('|')]))
def unroll_patterns(text: str,
patterns: List[Tuple[str, List[str]]],
label: Optional[str] = None) -> List[str]:
if not patterns:
return [text]
patterns = patterns.copy()
key, values = patterns.pop(0)
return (['// ' + label] if label else []) + list(
unroll_patterns(text.replace(key, value), patterns, value)
for value in values))
result = '\n'.join(unroll_patterns(text, patterns))
return result
def _expand_nonfinite(method: str, argstr: str, tail: str) -> str:
>>> print _expand_nonfinite('f', '<0 a>, <0 b>', ';')
f(a, 0);
f(0, b);
f(a, b);
>>> print _expand_nonfinite('f', '<0 a>, <0 b c>, <0 d>', ';')
f(a, 0, 0);
f(0, b, 0);
f(0, c, 0);
f(0, 0, d);
f(a, b, 0);
f(a, b, d);
f(a, 0, d);
f(0, b, d);
# argstr is "<valid-1 invalid1-1 invalid2-1 ...>, ..." (where usually
# 'invalid' is Infinity/-Infinity/NaN).
args = []
for arg in argstr.split(', '):
match = re.match('<(.*)>', arg)
if match is None:
raise InvalidTestDefinitionError(
f'Expected arg to match format "<(.*)>", but was: {arg}')
a =
args.append(a.split(' '))
calls = []
# Start with the valid argument list.
call = [args[j][0] for j in range(len(args))]
# For each argument alone, try setting it to all its invalid values:
for i, arg in enumerate(args):
for a in arg[1:]:
c2 = call[:]
c2[i] = a
# For all combinations of >= 2 arguments, try setting them to their
# first invalid values. (Don't do all invalid values, because the
# number of combinations explodes.)
def f(c: List[str], start: int, depth: int) -> None:
for i in range(start, len(args)):
if len(args[i]) > 1:
a = args[i][1]
c2 = c[:]
c2[i] = a
if depth > 0:
f(c2, i + 1, depth + 1)
f(call, 0, 0)
str_calls = (', '.join(c) for c in calls)
return '\n'.join(f'{method}({params}){tail}' for params in str_calls)
def _get_test_sub_dir(name: str, name_to_sub_dir: Mapping[str, str]) -> str:
for prefix in sorted(name_to_sub_dir.keys(), key=len, reverse=True):
if name.startswith(prefix):
return name_to_sub_dir[prefix]
raise InvalidTestDefinitionError(
f'Test "{name}" has no defined target directory mapping')
def _remove_extra_newlines(text: str) -> str:
"""Remove newlines if a backslash is found at end of line."""
# Lines ending with '\' gets their newline character removed.
text = re.sub(r'\\\n', '', text, flags=re.MULTILINE | re.DOTALL)
# Lines ending with '\-' gets their newline and any leading white spaces on
# the following line removed.
text = re.sub(r'\\-\n\s*', '', text, flags=re.MULTILINE | re.DOTALL)
return text
def _expand_test_code(code: str) -> str:
code = re.sub(r' @moz-todo', '', code)
code = re.sub(r'@moz-UniversalBrowserRead;', '', code)
code = _remove_extra_newlines(code)
# Unroll expressions with a cross-product-style parameter expansion.
code = re.sub(r'@unroll ([^;]*;)', lambda m: _unroll(, code)
code = re.sub(r'@nonfinite ([^(]+)\(([^)]+)\)(.*)', lambda m:
code) # Must come before '@assert throws'.
code = re.sub(r'@assert pixel (\d+,\d+) == (\d+,\d+,\d+,\d+);',
r'_assertPixel(canvas, \1, \2);', code)
code = re.sub(r'@assert pixel (\d+,\d+) ==~ (\d+,\d+,\d+,\d+);',
r'_assertPixelApprox(canvas, \1, \2, 2);', code)
code = re.sub(r'@assert pixel (\d+,\d+) ==~ (\d+,\d+,\d+,\d+) \+/- (\d+);',
r'_assertPixelApprox(canvas, \1, \2, \3);', code)
code = re.sub(r'@assert throws (\S+_ERR) (.*?);$',
r'assert_throws_dom("\1", function() { \2; });', code,
flags=re.MULTILINE | re.DOTALL)
code = re.sub(r'@assert throws (\S+Error) (.*?);$',
r'assert_throws_js(\1, function() { \2; });', code,
flags=re.MULTILINE | re.DOTALL)
code = re.sub(
r'@assert (.*) === (.*);', lambda m:
(f'_assertSame({}, {}, '
f'"{_escape_js(}", "{_escape_js(}");'), code)
code = re.sub(
r'@assert (.*) !== (.*);', lambda m:
(f'_assertDifferent({}, {}, '
f'"{_escape_js(}", "{_escape_js(}");'), code)
code = re.sub(
r'@assert (.*) =~ (.*);',
lambda m: f'assert_regexp_match({}, {});', code)
code = re.sub(
r'@assert (.*);',
lambda m: f'_assert({}, "{_escape_js(}");', code)
assert '@' not in code
return code
_TestParams = Mapping[str, Any]
_MutableTestParams = MutableMapping[str, Any]
class _CanvasType(str, enum.Enum):
HTML_CANVAS = 'HtmlCanvas'
OFFSCREEN_CANVAS = 'OffscreenCanvas'
WORKER = 'Worker'
class _TemplateType(str, enum.Enum):
REFERENCE = 'reference'
HTML_REFERENCE = 'html_reference'
TESTHARNESS = 'testharness'
class _OutputPaths:
element: str
offscreen: str
def sub_path(self, sub_dir: str):
"""Create a new _OutputPaths that is a subpath of this _OutputPath."""
return _OutputPaths(element=os.path.join(self.element, sub_dir),
offscreen=os.path.join(self.offscreen, sub_dir))
def _validate_test(test: _TestParams):
if test.get('expected', '') == 'green' and
r'@assert pixel .* 0,0,0,0;', test['code']):
print(f'Probable incorrect pixel test in {test["name"]}')
if 'size' in test and (not isinstance(test['size'], list)
or len(test['size']) != 2):
raise InvalidTestDefinitionError(
f'Invalid canvas size "{test["size"]}" in test {test["name"]}. '
'Expected an array with two numbers.')
if test['template_type'] == _TemplateType.TESTHARNESS:
valid_test_types = {'sync', 'async', 'promise'}
valid_test_types = {'promise'}
test_type = test.get('test_type')
if test_type is not None and test_type not in valid_test_types:
raise InvalidTestDefinitionError(
f'Invalid test_type: {test_type}. '
f'Valid values are: {valid_test_types}.')
def _render_template(jinja_env: jinja2.Environment, template: jinja2.Template,
params: _TestParams) -> str:
"""Renders the specified jinja template.
The template is repetitively rendered until no more changes are observed.
This allows for template parameters to refer to other template parameters.
rendered = template.render(params)
previous = ''
while rendered != previous:
previous = rendered
template = jinja_env.from_string(rendered)
rendered = template.render(params)
return rendered
def _render(jinja_env: jinja2.Environment, template_name: str,
params: _TestParams, output_file_name: str):
template = jinja_env.get_template(template_name)
file_content = _render_template(jinja_env, template, params)
pathlib.Path(output_file_name).write_text(file_content, 'utf-8')
def _preprocess_code(jinja_env: jinja2.Environment, code: str,
params: _TestParams) -> str:
code = _expand_test_code(code)
# Render the code on its own, as it could contain templates expanding
# to multiple lines. This is needed to get proper indentation of the
# code in the main template.
code = _render_template(jinja_env, jinja_env.from_string(code), params)
return code
class _Variant():
def __init__(self, params: _MutableTestParams) -> None:
self._params = params
def params(self) -> _TestParams:
"""Read-only getter for this variant's param dict."""
return self._params
def create_with_defaults(test: _TestParams) -> '_Variant':
"""Create a _Variant from the specified params.
Default values are added for certain parameters, if missing."""
params = {
'desc': '',
'size': [100, 50],
# Test name, which ultimately is used as filename. File variant
# dimension names are appended to this to produce unique filenames.
'name': '',
# List of this variant grid dimension names. This uniquely
# identifies a single variant in a variant grid file.
'grid_variant_names': [],
# List of this variant dimension names, including both file and grid
# dimensions.
'variant_names': [],
# Same as `variant_names`, but concatenated into a single string.
# This is a useful shorthand for tests having a single variant
# dimension.
'variant_name': '',
'images': [],
'svgimages': [],
'fonts': [],
return _Variant(params)
def merge_params(self, params: _TestParams) -> '_Variant':
"""Returns a new `_Variant` that merges `self.params` and `params`."""
new_params = copy.deepcopy(self._params)
return _Variant(new_params)
def _add_variant_name(self, name: str) -> None:
self._params['variant_name'] += (
('.' if self.params['variant_name'] else '') + name)
self._params['variant_names'] += [name]
def with_grid_variant_name(self, name: str) -> '_Variant':
"""Addend a variant name to include in the grid element label."""
self._params['grid_variant_names'] += [name]
return self
def with_file_variant_name(self, name: str) -> '_Variant':
"""Addend a variant name to include in the generated file name."""
if self.params.get('append_variants_to_name', True):
self._params['name'] += '.' + name
return self
def _render_param(self, jinja_env: jinja2.Environment,
param_name: str) -> None:
"""Render the specified parameter in-place in the `params` dict."""
value = self.params.get(param_name)
if value and isinstance(value, str):
self._params[param_name] = (
def _get_file_name(self) -> str:
file_name = self.params['name']
if 'manual' in self.params:
file_name += '-manual'
return file_name
def _get_canvas_types(self) -> FrozenSet[_CanvasType]:
canvas_types = self.params.get('canvas_types', _CanvasType)
invalid_types = {
for type in canvas_types if type not in list(_CanvasType)
if invalid_types:
raise InvalidTestDefinitionError(
f'Invalid canvas_types: {list(invalid_types)}. '
f'Accepted values are: {[t.value for t in _CanvasType]}')
return frozenset(_CanvasType(t) for t in canvas_types)
def _get_template_type(self) -> _TemplateType:
if 'reference' in self.params and 'html_reference' in self.params:
raise InvalidTestDefinitionError(
f'Test {self.params["name"]} is invalid, "reference" and '
'"html_reference" can\'t both be specified at the same time.')
if 'reference' in self.params:
return _TemplateType.REFERENCE
if 'html_reference' in self.params:
return _TemplateType.HTML_REFERENCE
return _TemplateType.TESTHARNESS
def finalize_params(self, jinja_env: jinja2.Environment,
variant_id: int) -> None:
"""Finalize this variant by adding computed param fields."""
self._params['id'] = variant_id
for param_name in ('name', 'desc', 'attributes'):
self._render_param(jinja_env, param_name)
self._params['file_name'] = self._get_file_name()
self._params['canvas_types'] = self._get_canvas_types()
self._params['template_type'] = self._get_template_type()
if 'reference' in self._params:
self._params['reference'] = _preprocess_code(
jinja_env, self._params['reference'], self._params)
if 'html_reference' in self._params:
self._params['html_reference'] = _preprocess_code(
jinja_env, self._params['html_reference'], self._params)
code_params = dict(self.params)
if _CanvasType.HTML_CANVAS in self.params['canvas_types']:
code_params['canvas_type'] = _CanvasType.HTML_CANVAS.value
self._params['code_element'] = _preprocess_code(
jinja_env, self._params['code'], code_params)
if _CanvasType.OFFSCREEN_CANVAS in self.params['canvas_types']:
code_params['canvas_type'] = _CanvasType.OFFSCREEN_CANVAS.value
self._params['code_offscreen'] = _preprocess_code(
jinja_env, self._params['code'], code_params)
if _CanvasType.WORKER in self.params['canvas_types']:
code_params['canvas_type'] = _CanvasType.WORKER.value
self._params['code_worker'] = _preprocess_code(
jinja_env, self._params['code'], code_params)
def generate_expected_image(self, output_dirs: _OutputPaths) -> None:
"""Creates a reference image using Cairo and save filename in params."""
if 'expected' not in self.params:
expected = self.params['expected']
name = self.params['name']
if expected == 'green':
self._params['expected_img'] = '/images/green-100x50.png'
if expected == 'clear':
self._params['expected_img'] = '/images/clear-100x50.png'
if ';' in expected:
print(f'Found semicolon in {name}')
expected = re.sub(
r'^size (\d+) (\d+)',
r'surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, \1, \2)'
r'\ncr = cairo.Context(surface)', expected)
output_paths = output_dirs.sub_path(name)
if _CanvasType.HTML_CANVAS in self.params['canvas_types']:
expected_canvas = (
eval(compile(expected_canvas, f'<test {name}>', 'exec'), {},
{'cairo': cairo})
if {_CanvasType.OFFSCREEN_CANVAS, _CanvasType.WORKER
} & self.params['canvas_types']:
expected_offscreen = (
eval(compile(expected_offscreen, f'<test {name}>', 'exec'), {},
{'cairo': cairo})
self._params['expected_img'] = f'{name}.png'
class _VariantGrid:
def __init__(self, variants: List[_Variant], grid_width: int) -> None:
self._variants = variants
self._grid_width = grid_width
self._file_name = None
self._canvas_types = None
self._template_type = None
self._params = None
def variants(self) -> List[_Variant]:
"""Read only getter for the list of variant in this grid."""
return self._variants
def file_name(self):
"""File name to which this grid will be written."""
if self._file_name is None:
self._file_name = self._unique_param('file_name')
return self._file_name
def canvas_types(self) -> FrozenSet[_CanvasType]:
"""Returns the set of all _CanvasType used by this grid's variants."""
if self._canvas_types is None:
self._canvas_types = self._param_set('canvas_types')
return self._canvas_types
def template_type(self) -> _TemplateType:
"""Returns the type of Jinja template needed to render this grid."""
if self._template_type is None:
self._template_type = self._unique_param('template_type')
return self._template_type
def params(self) -> _TestParams:
"""Returns this grid's param dict, used to render Jinja templates."""
if self._params is None:
if len(self.variants) == 1:
self._params = dict(self.variants[0].params)
self._params = self._get_grid_params()
return self._params
def finalize(self, jinja_env: jinja2.Environment):
"""Finalize this grid's variants, adding computed params fields."""
for variant_id, variant in enumerate(self.variants):
variant.finalize_params(jinja_env, variant_id)
def add_dimension(self, variants: Mapping[str,
_TestParams]) -> '_VariantGrid':
"""Adds a variant dimension to this variant grid.
If the grid currently has N variants, adding a dimension with M variants
results in a grid containing N*M variants. Of course, we can't display
more than 2 dimensions on a 2D screen, so adding dimensions beyond 2
repeats all previous dimensions down vertically, with the grid width
set to the number of variants of the first dimension (unless overridden
by setting `grid_width`). For instance, a 3D variant space with
dimensions 3 x 2 x 2 will result in this layout:
000 100 200
010 110 210
001 101 201
011 111 211
new_variants = [
old_variant.merge_params(params or {}).with_grid_variant_name(name)
for name, params in variants.items()
for old_variant in self.variants
# The first dimension dictates the grid-width, unless it was specified
# beforehand via the test params.
new_grid_width = (self._grid_width
if self._grid_width > 1 else len(variants))
return _VariantGrid(variants=new_variants, grid_width=new_grid_width)
def merge_params(self, name: str, params: _TestParams) -> '_VariantGrid':
"""Merges the specified `params` into every variant of this grid."""
return _VariantGrid(variants=[
for variant in self.variants
def _variants_for_canvas_type(
self, canvas_type: _CanvasType) -> List[_TestParams]:
"""Returns the variants of this grid enabled for `canvas_type`."""
return [
v.params for v in self.variants
if canvas_type in v.params['canvas_types']
def _unique_param(self, name: str) -> Any:
"""Returns the value of the `name` param for this grid.
All the variants in this grid must agree on the same value for this
parameter, or else an exception is thrown."""
values = {variant.params.get(name) for variant in self.variants}
if len(values) != 1:
raise InvalidTestDefinitionError(
'All variants in a variant grid must use the same value '
f'for property "{name}". Got these values: {values}. '
'Consider specifying the property outside of grid '
'variants dimensions (in the base test definition or in a '
'file variant dimension)')
return values.pop()
def _param_set(self, name: str):
"""Returns the set of all values this grid has for the `name` param.
The `name` parameter of each variant is expected to be a sequence.
These are all accumulated in a set and returned."""
return frozenset(sum([list(v.params[name]) for v in self.variants],
def _get_grid_params(self) -> _TestParams:
"""Returns the params dict needed to render this grid with Jinja."""
filter_variant = self._variants_for_canvas_type
grid_params = {
'element_variants': filter_variant(_CanvasType.HTML_CANVAS),
'offscreen_variants': filter_variant(_CanvasType.OFFSCREEN_CANVAS),
'worker_variants': filter_variant(_CanvasType.WORKER),
'grid_width': self._grid_width,
'name': self._unique_param('name'),
'test_type': self._unique_param('test_type'),
'fuzzy': self._unique_param('fuzzy'),
'timeout': self._unique_param('timeout'),
'notes': self._unique_param('notes'),
'images': self._param_set('images'),
'svgimages': self._param_set('svgimages'),
'fonts': self._param_set('fonts'),
if self.template_type in (_TemplateType.REFERENCE,
grid_params['desc'] = self._unique_param('desc')
return grid_params
def _write_reference_test(self, jinja_env: jinja2.Environment,
output_files: _OutputPaths):
grid = '_grid' if len(self.variants) > 1 else ''
# If variants don't all use the same offscreen and worker canvas types,
# the offscreen and worker grids won't be identical. The worker test
# therefore can't reuse the offscreen reference file.
offscreen_types = {_CanvasType.OFFSCREEN_CANVAS, _CanvasType.WORKER}
needs_worker_reference = len({
variant.params['canvas_types'] & offscreen_types
for variant in self.variants
}) != 1
params = dict(self.params)
params['reference_file'] = f'{params["name"]}-expected.html'
if _CanvasType.HTML_CANVAS in self.canvas_types:
_render(jinja_env, f'reftest_element{grid}.html', params,
if _CanvasType.OFFSCREEN_CANVAS in self.canvas_types:
_render(jinja_env, f'reftest_offscreen{grid}.html', params,
if _CanvasType.WORKER in self.canvas_types:
if needs_worker_reference:
params['reference_file'] = f'{params["name"]}.w-expected.html'
_render(jinja_env, f'reftest_worker{grid}.html', params,
params['is_test_reference'] = True
is_html_ref = self.template_type == _TemplateType.HTML_REFERENCE
ref_template_name = (f'reftest{grid}.html'
if is_html_ref else f'reftest_element{grid}.html')
if _CanvasType.HTML_CANVAS in self.canvas_types:
_render(jinja_env, ref_template_name, params,
if self.canvas_types & offscreen_types:
# We use the same template for all reference files, so we need to
# assign the variant definition to the variable expected by the
# template.
params['element_variants'] = params.get('offscreen_variants')
_render(jinja_env, ref_template_name, params,
if needs_worker_reference:
params['element_variants'] = params.get('worker_variants')
_render(jinja_env, ref_template_name, params,
def _write_testharness_test(self, jinja_env: jinja2.Environment,
output_files: _OutputPaths):
grid = '_grid' if len(self.variants) > 1 else ''
# Create test cases for canvas and offscreencanvas.
if _CanvasType.HTML_CANVAS in self.canvas_types:
_render(jinja_env, f'testharness_element{grid}.html', self.params,
if _CanvasType.OFFSCREEN_CANVAS in self.canvas_types:
_render(jinja_env, f'testharness_offscreen{grid}.html',
self.params, f'{output_files.offscreen}.html')
if _CanvasType.WORKER in self.canvas_types:
_render(jinja_env, f'testharness_worker{grid}.js', self.params,
def generate_test(self, jinja_env: jinja2.Environment,
output_dirs: _OutputPaths) -> None:
"""Generate the test files to the specified output dirs."""
output_files = output_dirs.sub_path(self.file_name)
if self.template_type in (_TemplateType.REFERENCE,
self._write_reference_test(jinja_env, output_files)
self._write_testharness_test(jinja_env, output_files)
class _VariantLayout(str, enum.Enum):
SINGLE_FILE = 'single_file'
MULTI_FILES = 'multi_files'
class _VariantDimension:
variants: Mapping[str, _TestParams]
layout: _VariantLayout
def _get_variant_dimensions(params: _TestParams) -> List[_VariantDimension]:
variants = params.get('variants', [])
if not isinstance(variants, list):
raise InvalidTestDefinitionError(
Variants must be specified as a list of variant dimensions, e.g.:
- dimension1-variant1:
param: ...
param: ...
- dimension2-variant1:
param: ...
param: ..."""))
variants_layout = params.get('variants_layout',
[_VariantLayout.MULTI_FILES] * len(variants))
if len(variants) != len(variants_layout):
raise InvalidTestDefinitionError(
'variants and variants_layout must be lists of the same size')
invalid_layouts = [
l for l in variants_layout if l not in list(_VariantLayout)
if invalid_layouts:
raise InvalidTestDefinitionError('Invalid variants_layout: ' +
', '.join(invalid_layouts) +
'. Valid layouts are: ' +
', '.join(_VariantLayout))
return [
_VariantDimension(z[0], z[1]) for z in zip(variants, variants_layout)
def _get_variant_grids(test: Mapping[str, Any]) -> List[_VariantGrid]:
base_variant = _Variant.create_with_defaults(test)
grid_width = base_variant.params.get('grid_width', 1)
grids = [_VariantGrid([base_variant], grid_width=grid_width)]
for dimension in _get_variant_dimensions(test):
variants = dimension.variants
if dimension.layout == _VariantLayout.MULTI_FILES:
grids = [
grid.merge_params(name, params)
for name, params in variants.items() for grid in grids
grids = [grid.add_dimension(variants) for grid in grids]
return grids
def _check_uniqueness(tested: DefaultDict[str, Set[_CanvasType]], name: str,
canvas_types: FrozenSet[_CanvasType]) -> None:
already_tested = tested[name].intersection(canvas_types)
if already_tested:
raise InvalidTestDefinitionError(
f'Test {name} is defined twice for types {already_tested}')
def generate_test_files(name_to_dir_file: str) -> None:
"""Generate Canvas tests from YAML file definition."""
output_dirs = _OutputPaths(element='../element', offscreen='../offscreen')
jinja_env = jinja2.Environment(
jinja_env.filters['double_quote_escape'] = _double_quote_escape
# Run with --test argument to run unit tests.
if len(sys.argv) > 1 and sys.argv[1] == '--test':
doctest = importlib.import_module('doctest')
name_to_sub_dir = (yaml.safe_load(
tests = []
test_yaml_directory = 'yaml-new'
yaml_files = [
os.path.join(test_yaml_directory, f)
for f in os.listdir(test_yaml_directory) if f.endswith('.yaml')
for t in sum([
for f in yaml_files
], []):
if 'DISABLED' in t:
if 'meta' in t:
eval(compile(t['meta'], '<meta test>', 'exec'), {},
{'tests': tests})
# Ensure the test output directories exist.
test_dirs = [output_dirs.element, output_dirs.offscreen]
for sub_dir in set(name_to_sub_dir.values()):
for d in test_dirs:
except FileExistsError:
pass # Ignore if it already exists,
used_filenames = collections.defaultdict(set)
used_variants = collections.defaultdict(set)
for test in tests:
for grid in _get_variant_grids(test):
if test['name'] != grid.file_name:
print(f' {grid.file_name}')
sub_dir = _get_test_sub_dir(grid.file_name, name_to_sub_dir)
output_sub_dirs = output_dirs.sub_path(sub_dir)
_check_uniqueness(used_filenames, grid.file_name,
for variant in grid.variants:
'.'.join([grid.file_name] +
for variant in grid.variants:
grid.generate_test(jinja_env, output_sub_dirs)