Updated to ACES 1.0.1

[OpenColorIO-Configs.git] / aces_1.0.1 / python / aces_ocio / tests / tests_aces_config.py

diff --git a/aces_1.0.1/python/aces_ocio/tests/tests_aces_config.py b/aces_1.0.1/python/aces_ocio/tests/tests_aces_config.py
new file mode 100644 (file)
index 0000000..fb7524a
--- /dev/null
+++ b/aces_1.0.1/python/aces_ocio/tests/tests_aces_config.py
@@ -0,0 +1,143 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+Defines unit tests for *ACES* configuration.
+"""
+
+from __future__ import division
+
+import hashlib
+import os
+import re
+import shutil
+import sys
+import tempfile
+import unittest
+
+sys.path.append(os.path.abspath(
+    os.path.join(os.path.dirname(__file__), '..', '..')))
+
+from aces_ocio.utilities import files_walker
+from aces_ocio.generate_config import (
+    ACES_OCIO_CTL_DIRECTORY_ENVIRON,
+    generate_config)
+
+__author__ = 'ACES Developers'
+__copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
+__license__ = ''
+__maintainer__ = 'ACES Developers'
+__email__ = 'aces@oscars.org'
+__status__ = 'Production'
+
+__all__ = ['REFERENCE_CONFIG_ROOT_DIRECTORY',
+           'HASH_TEST_PATTERNS',
+           'UNHASHABLE_TEST_PATTERNS',
+           'TestACESConfig']
+
+
+# TODO: Investigate how the current config has been generated to use it for
+# tests.
+REFERENCE_CONFIG_ROOT_DIRECTORY = os.path.abspath(
+    os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+HASH_TEST_PATTERNS = ('\.3dl', '\.lut', '\.csp')
+UNHASHABLE_TEST_PATTERNS = ('\.icc', '\.ocio')
+
+
+class TestACESConfig(unittest.TestCase):
+    """
+    Performs tests on the *ACES* configuration.
+    """
+
+    def setUp(self):
+        """
+        Initialises common tests attributes.
+        """
+
+        self.__aces_ocio_ctl_directory = os.environ.get(
+            ACES_OCIO_CTL_DIRECTORY_ENVIRON, None)
+
+        assert self.__aces_ocio_ctl_directory is not None, (
+            'Undefined "{0}" environment variable!'.format(
+                ACES_OCIO_CTL_DIRECTORY_ENVIRON))
+
+        assert os.path.exists(self.__aces_ocio_ctl_directory) is True, (
+            '"{0}" directory does not exists!'.format(
+                self.__aces_ocio_ctl_directory))
+
+        self.maxDiff = None
+        self.__temporary_directory = tempfile.mkdtemp()
+
+    def tearDown(self):
+        """
+        Post tests actions.
+        """
+
+        shutil.rmtree(self.__temporary_directory)
+
+    @staticmethod
+    def directory_hashes(directory,
+                         filters_in=None,
+                         filters_out=None,
+                         flags=0):
+        """
+        Recursively computes the hashes from the file within given directory.
+
+        Parameters
+        ----------
+        directory : str or unicode
+            Directory to compute the file hashes.
+        filters_in : array_like
+            Included patterns.
+        filters_out : array_like
+            Excluded patterns.
+        flags : int
+            Regex flags.
+
+        Returns
+        -------
+        dict
+             Directory file hashes.
+        """
+
+        hashes = {}
+        for path in files_walker(directory,
+                                 filters_in=filters_in,
+                                 filters_out=filters_out,
+                                 flags=flags):
+            with open(path) as file:
+                digest = hashlib.md5(
+                    re.sub('\s', '', file.read())).hexdigest()
+            hashes[path.replace(directory, '')] = digest
+        return hashes
+
+    def test_ACES_config(self):
+        """
+        Performs tests on the *ACES* configuration by computing hashes on the
+        generated configuration and comparing them to the existing one.
+        """
+
+        self.assertTrue(generate_config(self.__aces_ocio_ctl_directory,
+                                        self.__temporary_directory))
+
+        reference_hashes = self.directory_hashes(
+            REFERENCE_CONFIG_ROOT_DIRECTORY,
+            HASH_TEST_PATTERNS)
+        test_hashes = self.directory_hashes(
+            self.__temporary_directory,
+            HASH_TEST_PATTERNS)
+
+        self.assertDictEqual(reference_hashes, test_hashes)
+
+        # Checking that unashable files ('.icc', '.ocio') are generated.
+        unashable = lambda x: (
+            sorted([file.replace(x, '') for file in
+                    files_walker(x, UNHASHABLE_TEST_PATTERNS)]))
+
+        self.assertListEqual(unashable(REFERENCE_CONFIG_ROOT_DIRECTORY),
+                             unashable(self.__temporary_directory))
+
+
+if __name__ == '__main__':
+    unittest.main()