expose ImageCms.core.CmsTransform

nulano · Jan 20, 2024 · 1e8e186 · 1e8e186
1 parent 1bfa30a
commit 1e8e186
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Tests/test_imagecms.py b/Tests/test_imagecms.py
@@ -512,6 +512,14 @@ def test_profile_typesafety():
         ImageCms.core.CmsProfile(0)
 
 
+def test_transform_typesafety():
+    # core transform should not be directly instantiable
+    with pytest.raises(TypeError):
+        ImageCms.core.CmsProfile()
+    with pytest.raises(TypeError):
+        ImageCms.core.CmsProfile(0)
+
+
 def assert_aux_channel_preserved(mode, transform_in_place, preserved_channel):
     def create_test_image():
         # set up test image with something interesting in the tested aux channel.

diff --git a/src/_imagingcms.c b/src/_imagingcms.c
@@ -1518,6 +1518,9 @@ setup_module(PyObject *m) {
     Py_INCREF(&CmsProfile_Type);
     PyModule_AddObject(m, "CmsProfile", (PyObject *)&CmsProfile_Type);
 
+    Py_INCREF(&CmsTransform_Type);
+    PyModule_AddObject(m, "CmsTransform", (PyObject *)&CmsTransform_Type);
+
     d = PyModule_GetDict(m);
 
     /* this check is also in PIL.features.pilinfo() */