Implement support for continuous function iteration.

Author: KelSolaar

colour/continuous/abstract.py

@@ -19,6 +19,7 @@
     ArrayLike,
     Callable,
     DTypeFloat,
+    Generator,
     Literal,
     NDArrayFloat,
     Optional,
@@ -96,6 +97,7 @@ class AbstractContinuousFunction(ABC):
     -   :meth:`~colour.continuous.AbstractContinuousFunction.__getitem__`
     -   :meth:`~colour.continuous.AbstractContinuousFunction.__setitem__`
     -   :meth:`~colour.continuous.AbstractContinuousFunction.__contains__`
+    -   :meth:`~colour.continuous.AbstractContinuousFunction.__iter__`
     -   :meth:`~colour.continuous.AbstractContinuousFunction.__len__`
     -   :meth:`~colour.continuous.AbstractContinuousFunction.__eq__`
     -   :meth:`~colour.continuous.AbstractContinuousFunction.__ne__`
@@ -484,6 +486,18 @@ def __contains__(self, x: Union[ArrayLike, slice]) -> bool:
 
         ...  # pragma: no cover
 
+    def __iter__(self) -> Generator:
+        """
+        Return a generator for the abstract continuous function.
+
+        Yields
+        ------
+        Generator
+           Abstract continuous function generator.
+        """
+
+        yield from np.column_stack([self.domain, self.range])
+
     def __len__(self) -> int:
         """
         Return the abstract continuous function independent domain :math:`x`

colour/continuous/tests/test_abstract.py

@@ -51,6 +51,7 @@ def test_required_methods(self):
             "__getitem__",
             "__setitem__",
             "__contains__",
+            "__iter__",
             "__len__",
             "__eq__",
             "__ne__",

colour/continuous/tests/test_multi_signal.py

@@ -79,6 +79,7 @@ def test_required_methods(self):
             "__getitem__",
             "__setitem__",
             "__contains__",
+            "__iter__",
             "__eq__",
             "__ne__",
             "arithmetical_operation",
@@ -745,6 +746,16 @@ def test__contains__(self):
         self.assertIn(0.5, self._multi_signals)
         self.assertNotIn(1000, self._multi_signals)
 
+    def test__iter__(self):
+        """Test :func:`colour.continuous.signal.Signal.__iter__` method."""
+
+        domain = np.arange(0, 10)
+        for i, domain_range_value in enumerate(self._multi_signals):
+            np.testing.assert_array_equal(domain_range_value[0], domain[i])
+            np.testing.assert_array_equal(
+                domain_range_value[1:], self._range_2[i]
+            )
+
     def test__len__(self):
         """
         Test :meth:`colour.continuous.multi_signals.MultiSignals.__len__`

colour/continuous/tests/test_signal.py

@@ -65,6 +65,7 @@ def test_required_methods(self):
             "__getitem__",
             "__setitem__",
             "__contains__",
+            "__iter__",
             "__eq__",
             "__ne__",
             "arithmetical_operation",
@@ -477,6 +478,14 @@ def test__contains__(self):
         self.assertIn(0.5, self._signal)
         self.assertNotIn(1000, self._signal)
 
+    def test__iter__(self):
+        """Test :func:`colour.continuous.signal.Signal.__iter__` method."""
+
+        domain = np.arange(0, 10)
+        for i, (domain_value, range_value) in enumerate(self._signal):
+            np.testing.assert_array_equal(domain_value, domain[i])
+            np.testing.assert_array_equal(range_value, self._range[i])
+
     def test__len__(self):
         """Test :func:`colour.continuous.signal.Signal.__len__` method."""