1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
# SPDX-License-Identifier: Apache-2.0
# -----------------------------------------------------------------------------
# Copyright 2020 Arm Limited
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
# use this file except in compliance with the License. You may obtain a copy
# of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
# -----------------------------------------------------------------------------
"""
An ASTC TestSet is comprised of a set of TestImages. Images are stored in a
structured directory layout. This structure encodes important metadata about
each image - such as color profile and data encoding - in the directory and
file names used.
TestSets are built by using reflection on a root directory to automatically
find all of the test images that comprise the set.
"""
import os
from testlib.image import TestImage
class TSetException(Exception):
"""
Exception thrown for bad test set specification.
"""
class TestSet():
"""
Generate a list of images that are test candidates.
This reflection is built automatically based on a directory of images on
disk, provided that the images follow a standard structure.
Attributes:
name: The name of the test set.
tests: The list of TestImages forming the set.
"""
def __init__(self, name, rootDir, profiles, formats, imageFilter=None):
"""
Create a new TestSet through reflection.
Args:
name (str): The name of the test set.
rootDir (str): The root directory of the test set.
profiles (list(str)): The ASTC profiles to allow.
formats (list(str)): The image formats to allow.
imageFilter (str): The name of the image to include (for bug repo).
Raises:
TSetException: The specified TestSet could not be loaded.
"""
self.name = name
if not os.path.exists(rootDir) and not os.path.isdir(rootDir):
raise TSetException("Bad test set root directory (%s)" % rootDir)
self.tests = []
for (dirPath, dirNames, fileNames) in os.walk(rootDir):
for fileName in fileNames:
# Only select image files
fileExt = os.path.splitext(fileName)[1]
if fileExt not in TestImage.TEST_EXTS:
continue
# Create the TestImage for each file on disk
filePath = os.path.join(dirPath, fileName)
image = TestImage(filePath)
# Filter out the ones we don't want to allow
if image.colorProfile not in profiles:
continue
if image.colorFormat not in formats:
continue
if imageFilter and image.testFile != imageFilter:
continue
self.tests.append((filePath, image))
# Sort the TestImages so they are in a stable order
self.tests.sort()
self.tests = [x[1] for x in self.tests]
|
