OpenColorIO/tests/python/test_BuiltinTransform.py at ad086487ffc740e3702f2796fe2bea5d4e675b91 · AcademySoftwareFoundation/OpenColorIO · GitHub

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
# SPDX-License-Identifier: BSD-3-Clause
# Copyright Contributors to the OpenColorIO Project.

import unittest

import PyOpenColorIO as OCIO
from test_TransformsBase import TransformsBaseTest


class TestBuiltinTransform(unittest.TestCase, TransformsBaseTest):
    # BuiltinTransformRegistry singleton
    REGISTRY = None

    # Default values
    DEFAULT_STYLE = 'IDENTITY'
    DEFAULT_DESC = ''
    DEFAULT_STR = '<BuiltinTransform direction = forward, style = IDENTITY>'

    # One reference style
    EXAMPLE_STYLE = 'UTILITY - ACES-AP0_to_CIE-XYZ-D65_BFD'
    EXAMPLE_DESC = (
        'Convert ACES AP0 primaries to CIE XYZ with a D65 white point with '
        'Bradford adaptation')

    @classmethod
    def setUpClass(cls):
        cls.REGISTRY = OCIO.BuiltinTransformRegistry()

    def setUp(self):
        self.tr = OCIO.BuiltinTransform()

    def test_transform_type(self):
        # TransformType is correct
        self.assertEqual(self.tr.getTransformType(), OCIO.TRANSFORM_TYPE_BUILTIN)

    def test_style(self):
        # Default style is identity
        self.assertEqual(self.tr.getStyle(), self.DEFAULT_STYLE)
        self.assertEqual(self.tr.getDescription(), self.DEFAULT_DESC)

        # Set/get all styles from registry
        for style, desc in self.REGISTRY.getBuiltins():
            self.tr.setStyle(style)
            self.assertEqual(self.tr.getStyle(), style)
            self.assertEqual(self.tr.getDescription(), desc)
            self.assertIsNone(self.tr.validate())  # Does not raise

        # Invalid style raises OCIO Exception
        with self.assertRaises(OCIO.Exception):
            self.tr.setStyle("invalid")

        # Safe invalid type handling
        for invalid in (None, 1, True):
            with self.assertRaises(TypeError):
                self.tr.setStyle(invalid)

    def test_constructor_keyword(self):
        # Keyword args in order
        builtin_tr1 = OCIO.BuiltinTransform(style=self.EXAMPLE_STYLE,
                                            direction=OCIO.TRANSFORM_DIR_FORWARD)

        self.assertEqual(builtin_tr1.getStyle(), self.EXAMPLE_STYLE)
        self.assertEqual(builtin_tr1.getDescription(), self.EXAMPLE_DESC)
        self.assertEqual(builtin_tr1.getDirection(),
                         OCIO.TRANSFORM_DIR_FORWARD)

        # Keyword args out of order
        builtin_tr2 = OCIO.BuiltinTransform(direction=OCIO.TRANSFORM_DIR_FORWARD,
                                            style=self.EXAMPLE_STYLE)

        self.assertEqual(builtin_tr2.getStyle(), self.EXAMPLE_STYLE)
        self.assertEqual(builtin_tr2.getDescription(), self.EXAMPLE_DESC)
        self.assertEqual(builtin_tr2.getDirection(),
                         OCIO.TRANSFORM_DIR_FORWARD)

    def test_constructor_positional(self):
        # Positional args
        builtin_tr = OCIO.BuiltinTransform(self.EXAMPLE_STYLE,
                                           OCIO.TRANSFORM_DIR_FORWARD)

        self.assertEqual(builtin_tr.getStyle(), self.EXAMPLE_STYLE)
        self.assertEqual(builtin_tr.getDescription(), self.EXAMPLE_DESC)
        self.assertEqual(builtin_tr.getDirection(),
                         OCIO.TRANSFORM_DIR_FORWARD)

    def test_str(self):
        # str() and print() result matches C++ << operator
        self.assertEqual(str(self.tr), self.DEFAULT_STR)