def setUp(self):
"""Initialise the common tests attributes."""
self._range_1 = np.linspace(10, 100, 10)
self._range_2 = tstack([self._range_1] * 3) + np.array([0, 10, 20])
self._domain_1 = np.arange(0, 10, 1)
self._domain_2 = np.arange(100, 1100, 100)
self._multi_signals = MultiSignals(self._range_2)
def test_raise_exception__contains__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__contains__`
method raised exception.
"""
self.assertRaises(RuntimeError, operator.contains, MultiSignals(), 0)
def test__repr__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__repr__`
method.
"""
self.assertEqual(
re.sub(r'extrapolator_args={.*}', 'extrapolator_args={...}',
repr(self._multi_signals)),
textwrap.dedent("""
MultiSignals([[ 0., 10., 20., 30.],
[ 1., 20., 30., 40.],
[ 2., 30., 40., 50.],
[ 3., 40., 50., 60.],
[ 4., 50., 60., 70.],
[ 5., 60., 70., 80.],
[ 6., 70., 80., 90.],
[ 7., 80., 90., 100.],
[ 8., 90., 100., 110.],
[ 9., 100., 110., 120.]],
labels=[0, 1, 2],
interpolator=KernelInterpolator,
interpolator_args={},
extrapolator=Extrapolator,
extrapolator_args={...})""")[1:])
self.assertIsInstance(repr(MultiSignals()), string_types)
def test__repr__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__repr__`
method.
"""
self.assertEqual(
re.sub(
r"extrapolator_kwargs={.*}",
"extrapolator_kwargs={...}",
repr(self._multi_signals),
),
textwrap.dedent("""
MultiSignals([[ 0., 10., 20., 30.],
[ 1., 20., 30., 40.],
[ 2., 30., 40., 50.],
[ 3., 40., 50., 60.],
[ 4., 50., 60., 70.],
[ 5., 60., 70., 80.],
[ 6., 70., 80., 90.],
[ 7., 80., 90., 100.],
[ 8., 90., 100., 110.],
[ 9., 100., 110., 120.]],
labels=['0', '1', '2'],
interpolator=KernelInterpolator,
interpolator_kwargs={},
extrapolator=Extrapolator,
extrapolator_kwargs={...})""")[1:],
)
self.assertIsInstance(repr(MultiSignals()), str)
def test_raise_exception_function(self):
"""
Test :func:`colour.continuous.signal.multi_signals.MultiSignals.\
function` property raised exception.
"""
self.assertRaises((ValueError, TypeError), MultiSignals().function, 0)
def test_raise_exception_function(self):
"""
Tests :func:`colour.continuous.signal.multi_signals.MultiSignals`
property raised exception.
"""
self.assertRaises(RuntimeError, MultiSignals().function, 0)
def test_multi_signals_unpack_data(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.\
multi_signals_unpack_data` method.
"""
signals = MultiSignals.multi_signals_unpack_data(self._range_1)
self.assertListEqual(list(signals.keys()), [0])
np.testing.assert_array_equal(signals[0].domain, self._domain_1)
np.testing.assert_array_equal(signals[0].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_1, self._domain_2)
self.assertListEqual(list(signals.keys()), [0])
np.testing.assert_array_equal(signals[0].domain, self._domain_2)
np.testing.assert_array_equal(signals[0].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_2, self._domain_2)
self.assertListEqual(list(signals.keys()), [0, 1, 2])
np.testing.assert_array_equal(signals[0].range, self._range_1)
np.testing.assert_array_equal(signals[1].range, self._range_1 + 10)
np.testing.assert_array_equal(signals[2].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2)))
self.assertListEqual(list(signals.keys()), [0, 1, 2])
np.testing.assert_array_equal(signals[0].range, self._range_1)
np.testing.assert_array_equal(signals[1].range, self._range_1 + 10)
np.testing.assert_array_equal(signals[2].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))))
self.assertListEqual(list(signals.keys()), [0, 1, 2])
np.testing.assert_array_equal(signals[0].range, self._range_1)
np.testing.assert_array_equal(signals[1].range, self._range_1 + 10)
np.testing.assert_array_equal(signals[2].range, self._range_1 + 20)
if is_pandas_installed():
from pandas import DataFrame, Series
signals = MultiSignals.multi_signals_unpack_data(
Series(dict(zip(self._domain_1, self._range_1))))
self.assertListEqual(list(signals.keys()), [0])
np.testing.assert_array_equal(signals[0].domain, self._domain_1)
np.testing.assert_array_equal(signals[0].range, self._range_1)
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
signals = MultiSignals.multi_signals_unpack_data(
DataFrame(data, self._domain_1))
self.assertListEqual(list(signals.keys()), ['a', 'b', 'c'])
np.testing.assert_array_equal(signals['a'].range, self._range_1)
np.testing.assert_array_equal(signals['b'].range,
self._range_1 + 10)
np.testing.assert_array_equal(signals['c'].range,
self._range_1 + 20)
def test_signal_type(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.signal_type`
property.
"""
multi_signals = MultiSignals(signal_type=Signal)
self.assertEqual(multi_signals.signal_type, Signal)
def test__init__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__init__`
method.
"""
multi_signals = MultiSignals(self._range_1)
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
multi_signals = MultiSignals(self._range_1, self._domain_2)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
multi_signals = MultiSignals(self._range_2, self._domain_2)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
multi_signals = MultiSignals(dict(zip(self._domain_2, self._range_2)))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
multi_signals = MultiSignals(multi_signals)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
class NotSignal(Signal):
"""
Not :class:`Signal` class.
"""
pass
multi_signals = MultiSignals(self._range_1, signal_type=NotSignal)
self.assertIsInstance(multi_signals.signals[0], NotSignal)
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
if is_pandas_installed():
from pandas import DataFrame, Series
multi_signals = MultiSignals(
Series(dict(zip(self._domain_2, self._range_1))))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
multi_signals = MultiSignals(DataFrame(data, self._domain_2))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
def test_to_dataframe(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.to_dataframe`
method.
"""
if is_pandas_installed():
from pandas import DataFrame
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
assert MultiSignals(self._range_2,
self._domain_2,
labels=['a', 'b', 'c']).to_dataframe().equals(
DataFrame(data, self._domain_2))
def test_to_dataframe(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.to_dataframe`
method.
"""
if is_pandas_installed():
from pandas import DataFrame
data = dict(zip(["a", "b", "c"], tsplit(self._range_2)))
attest(
MultiSignals(self._range_2,
self._domain_2,
labels=["a", "b", "c"]).to_dataframe().equals(
DataFrame(data, self._domain_2)))
def read(self):
"""
Reads and parses the spectral data JSON file path.
Returns
-------
bool
Definition success.
"""
with open(self._path, 'r') as json_file:
content = json.load(json_file)
self._header = SpectralDataHeader_AMPAS(**content['header'])
for attribute in ('units', 'reflection_geometry',
'transmission_geometry', 'bandwidth_FWHM',
'bandwidth_corrected'):
setattr(self, '_{0}'.format(attribute),
content['spectral_data'][attribute])
index = content['spectral_data']['index']['main']
data = content['spectral_data']['data']['main']
if len(index) == 1:
self.domain, self.range = Signal.signal_unpack_data(
{k: v[0]
for k, v in data.items()})
else:
self.signals = MultiSignals.multi_signals_unpack_data(data,
labels=index)
# TODO: Re-instate "manufacturer", "model", "illuminant" and "type"
# attributes according to outcome of
# https://github.com/ampas/rawtoaces/issues/114.
if ('manufacturer' in self._header._kwargs
and 'model' in self._header._kwargs):
self.name = '{0} {1}'.format(self._header._kwargs['manufacturer'],
self._header._kwargs['model'])
elif 'illuminant' in self._header._kwargs:
self.name = self._header._kwargs['illuminant']
elif 'type' in self._header._kwargs:
self.name = self._header._kwargs['type']
return self
def test__str__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__str__`
method.
"""
self.assertEqual(
str(self._multi_signals),
textwrap.dedent("""
[[ 0. 10. 20. 30.]
[ 1. 20. 30. 40.]
[ 2. 30. 40. 50.]
[ 3. 40. 50. 60.]
[ 4. 50. 60. 70.]
[ 5. 60. 70. 80.]
[ 6. 70. 80. 90.]
[ 7. 80. 90. 100.]
[ 8. 90. 100. 110.]
[ 9. 100. 110. 120.]]""")[1:])
self.assertIsInstance(str(MultiSignals()), string_types)
class TestMultiSignals(unittest.TestCase):
"""
Defines :class:`colour.continuous.multi_signals.MultiSignals` class unit
tests methods.
"""
def setUp(self):
"""
Initialises common tests attributes.
"""
self._range_1 = np.linspace(10, 100, 10)
self._range_2 = tstack([self._range_1] * 3) + np.array([0, 10, 20])
self._domain_1 = np.arange(0, 10, 1)
self._domain_2 = np.arange(100, 1100, 100)
self._multi_signals = MultiSignals(self._range_2)
def test_required_attributes(self):
"""
Tests presence of required attributes.
"""
required_attributes = ('dtype', 'domain', 'range', 'interpolator',
'interpolator_args', 'extrapolator',
'extrapolator_args', 'function', 'signals',
'labels', 'signal_type')
for attribute in required_attributes:
self.assertIn(attribute, dir(MultiSignals))
def test_required_methods(self):
"""
Tests presence of required methods.
"""
required_methods = ('__str__', '__repr__', '__hash__', '__getitem__',
'__setitem__', '__contains__', '__eq__', '__ne__',
'arithmetical_operation',
'multi_signals_unpack_data', 'fill_nan',
'domain_distance', 'to_dataframe')
for method in required_methods:
self.assertIn(method, dir(MultiSignals))
def test_dtype(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.dtype`
property.
"""
self.assertEqual(self._multi_signals.dtype, None)
multi_signals = self._multi_signals.copy()
multi_signals.dtype = DEFAULT_FLOAT_DTYPE
self.assertEqual(multi_signals.dtype, DEFAULT_FLOAT_DTYPE)
def test_domain(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.domain`
property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]),
decimal=7)
multi_signals.domain = self._domain_1 * 10
np.testing.assert_array_equal(multi_signals.domain,
self._domain_1 * 10)
np.testing.assert_almost_equal(multi_signals[np.array([0, 1, 2]) * 10],
np.array([[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]),
decimal=7)
# TODO: Use "assertWarns" when dropping Python 2.7.
domain = np.linspace(0, 1, 10)
domain[0] = -np.inf
multi_signals.domain = domain
def test_range(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.range`
property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]),
decimal=7)
multi_signals.range = self._range_1 * 10
np.testing.assert_array_equal(multi_signals.range,
tstack([self._range_1] * 3) * 10)
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 10.0, 10.0], [20.0, 20.0, 20.0],
[30.0, 30.0, 30.0]]) * 10,
decimal=7)
multi_signals.range = self._range_2 * 10
np.testing.assert_array_equal(multi_signals.range, self._range_2 * 10)
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 20.0, 30.0], [20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]) * 10,
decimal=7)
def test_interpolator(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.interpolator`
property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([[10.00000000, 20.00000000, 30.00000000],
[22.83489024, 32.80460562, 42.77432100],
[34.80044921, 44.74343470, 54.68642018],
[47.55353925, 57.52325463, 67.49297001],
[60.00000000, 70.00000000, 80.00000000]]),
decimal=7)
multi_signals.interpolator = CubicSplineInterpolator
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([[10.00000000, 20.00000000, 30.00000000],
[22.50000000, 32.50000000, 42.50000000],
[35.00000000, 45.00000000, 55.00000000],
[47.50000000, 57.50000000, 67.50000000],
[60.00000000, 70.00000000, 80.00000000]]),
decimal=7)
def test_interpolator_args(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.\
interpolator_args` property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([[10.00000000, 20.00000000, 30.00000000],
[22.83489024, 32.80460562, 42.77432100],
[34.80044921, 44.74343470, 54.68642018],
[47.55353925, 57.52325463, 67.49297001],
[60.00000000, 70.00000000, 80.00000000]]),
decimal=7)
multi_signals.interpolator_args = {
'window': 1,
'kernel_args': {
'a': 1
}
}
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([[10.00000000, 20.00000000, 30.00000000],
[18.91328761, 27.91961505, 36.92594248],
[28.36993142, 36.47562611, 44.58132080],
[44.13100443, 53.13733187, 62.14365930],
[60.00000000, 70.00000000, 80.00000000]]),
decimal=7)
def test_extrapolator(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.extrapolator`
property.
"""
self.assertIsInstance(self._multi_signals.extrapolator(), Extrapolator)
def test_extrapolator_args(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.\
extrapolator_args` property.
"""
multi_signals = self._multi_signals.copy()
assert np.all(np.isnan(multi_signals[np.array([-1000, 1000])]))
multi_signals.extrapolator_args = {
'method': 'Linear',
}
np.testing.assert_almost_equal(multi_signals[np.array([-1000, 1000])],
np.array([[-9990.0, -9980.0, -9970.0],
[10010.0, 10020.0, 10030.0]]),
decimal=7)
def test_function(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.function`
property.
"""
assert hasattr(self._multi_signals.function, '__call__')
def test_raise_exception_function(self):
"""
Tests :func:`colour.continuous.signal.multi_signals.MultiSignals`
property raised exception.
"""
self.assertRaises((RuntimeError, TypeError),
MultiSignals().function, 0)
def test_signals(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.signals`
property.
"""
multi_signals = self._multi_signals.copy()
multi_signals.signals = self._range_1
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
def test_labels(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.labels`
property.
"""
self.assertListEqual(self._multi_signals.labels, [0, 1, 2])
multi_signals = self._multi_signals.copy()
multi_signals.labels = ['a', 'b', 'c']
self.assertListEqual(multi_signals.labels, ['a', 'b', 'c'])
def test_signal_type(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.signal_type`
property.
"""
multi_signals = MultiSignals(signal_type=Signal)
self.assertEqual(multi_signals.signal_type, Signal)
def test__init__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__init__`
method.
"""
multi_signals = MultiSignals(self._range_1)
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
multi_signals = MultiSignals(self._range_1, self._domain_2)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
multi_signals = MultiSignals(self._range_2, self._domain_2)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
multi_signals = MultiSignals(dict(zip(self._domain_2, self._range_2)))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
multi_signals = MultiSignals(multi_signals)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
class NotSignal(Signal):
"""
Not :class:`Signal` class.
"""
pass
multi_signals = MultiSignals(self._range_1, signal_type=NotSignal)
self.assertIsInstance(multi_signals.signals[0], NotSignal)
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
if is_pandas_installed():
from pandas import DataFrame, Series
multi_signals = MultiSignals(
Series(dict(zip(self._domain_2, self._range_1))))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
multi_signals = MultiSignals(DataFrame(data, self._domain_2))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
def test__hash__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__hash__`
method.
"""
self.assertIsInstance(hash(self._multi_signals), int)
def test__str__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__str__`
method.
"""
self.assertEqual(
str(self._multi_signals),
textwrap.dedent("""
[[ 0. 10. 20. 30.]
[ 1. 20. 30. 40.]
[ 2. 30. 40. 50.]
[ 3. 40. 50. 60.]
[ 4. 50. 60. 70.]
[ 5. 60. 70. 80.]
[ 6. 70. 80. 90.]
[ 7. 80. 90. 100.]
[ 8. 90. 100. 110.]
[ 9. 100. 110. 120.]]""")[1:])
self.assertIsInstance(str(MultiSignals()), string_types)
def test__repr__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__repr__`
method.
"""
self.assertEqual(
re.sub(r'extrapolator_args={.*}', 'extrapolator_args={...}',
repr(self._multi_signals)),
textwrap.dedent("""
MultiSignals([[ 0., 10., 20., 30.],
[ 1., 20., 30., 40.],
[ 2., 30., 40., 50.],
[ 3., 40., 50., 60.],
[ 4., 50., 60., 70.],
[ 5., 60., 70., 80.],
[ 6., 70., 80., 90.],
[ 7., 80., 90., 100.],
[ 8., 90., 100., 110.],
[ 9., 100., 110., 120.]],
labels=[0, 1, 2],
interpolator=KernelInterpolator,
interpolator_args={},
extrapolator=Extrapolator,
extrapolator_args={...})""")[1:])
self.assertIsInstance(repr(MultiSignals()), string_types)
def test__getitem__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__getitem__`
method.
"""
np.testing.assert_almost_equal(self._multi_signals[0],
np.array([10.0, 20.0, 30.0]),
decimal=7)
np.testing.assert_almost_equal(self._multi_signals[np.array([0, 1,
2])],
np.array([[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]),
decimal=7)
np.testing.assert_almost_equal(
self._multi_signals[np.linspace(0, 5, 5)],
np.array([[10.00000000, 20.00000000, 30.00000000],
[22.83489024, 32.80460562, 42.77432100],
[34.80044921, 44.74343470, 54.68642018],
[47.55353925, 57.52325463, 67.49297001],
[60.00000000, 70.00000000, 80.00000000]]),
decimal=7)
assert np.all(np.isnan(self._multi_signals[np.array([-1000, 1000])]))
multi_signals = self._multi_signals.copy()
multi_signals.extrapolator_args = {
'method': 'Linear',
}
np.testing.assert_array_equal(
multi_signals[np.array([-1000, 1000])],
np.array([[-9990.0, -9980.0, -9970.0], [10010.0, 10020.0,
10030.0]]))
multi_signals.extrapolator_args = {
'method': 'Constant',
'left': 0,
'right': 1
}
np.testing.assert_array_equal(
multi_signals[np.array([-1000, 1000])],
np.array([[0.0, 0.0, 0.0], [1.0, 1.0, 1.0]]))
def test_raise_exception__getitem__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__getitem__`
method raised exception.
"""
self.assertRaises(RuntimeError, operator.getitem, MultiSignals(), 0)
def test__setitem__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__setitem__`
method.
"""
multi_signals = self._multi_signals.copy()
multi_signals[0] = 20
np.testing.assert_almost_equal(multi_signals[0],
np.array([20.0, 20.0, 20.0]),
decimal=7)
multi_signals[np.array([0, 1, 2])] = 30
np.testing.assert_almost_equal(multi_signals[np.array([0, 1, 2])],
np.array([[30.0, 30.0, 30.0],
[30.0, 30.0, 30.0],
[30.0, 30.0, 30.0]]),
decimal=7)
multi_signals[0:3] = 40
np.testing.assert_almost_equal(multi_signals[0:3],
np.array([[40.0, 40.0, 40.0],
[40.0, 40.0, 40.0],
[40.0, 40.0, 40.0]]),
decimal=7)
multi_signals[np.linspace(0, 5, 5)] = 50
np.testing.assert_almost_equal(multi_signals.domain,
np.array([
0.00, 1.00, 1.25, 2.00, 2.50, 3.00,
3.75, 4.00, 5.00, 6.00, 7.00, 8.00,
9.00
]),
decimal=7)
np.testing.assert_almost_equal(multi_signals.range,
np.array([
[50.0, 50.0, 50.0],
[40.0, 40.0, 40.0],
[50.0, 50.0, 50.0],
[40.0, 40.0, 40.0],
[50.0, 50.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 50.0, 50.0],
[50.0, 60.0, 70.0],
[50.0, 50.0, 50.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7)
multi_signals[np.array([0, 1, 2])] = np.array([10, 20, 30])
np.testing.assert_almost_equal(multi_signals.range,
np.array([
[10.0, 20.0, 30.0],
[10.0, 20.0, 30.0],
[50.0, 50.0, 50.0],
[10.0, 20.0, 30.0],
[50.0, 50.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 50.0, 50.0],
[50.0, 60.0, 70.0],
[50.0, 50.0, 50.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7)
multi_signals[np.array([0, 1, 2])] = np.arange(1, 10, 1).reshape(3, 3)
np.testing.assert_almost_equal(multi_signals.range,
np.array([
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
[50.0, 50.0, 50.0],
[7.0, 8.0, 9.0],
[50.0, 50.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 50.0, 50.0],
[50.0, 60.0, 70.0],
[50.0, 50.0, 50.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7)
def test__contains__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__contains__`
method.
"""
self.assertIn(0, self._multi_signals)
self.assertIn(0.5, self._multi_signals)
self.assertNotIn(1000, self._multi_signals)
def test_raise_exception__contains__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__contains__`
method raised exception.
"""
self.assertRaises(RuntimeError, operator.contains, MultiSignals(), 0)
def test__len__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__len__`
method.
"""
self.assertEqual(len(self._multi_signals), 10)
def test__eq__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__eq__`
method.
"""
signal_1 = self._multi_signals.copy()
signal_2 = self._multi_signals.copy()
self.assertEqual(signal_1, signal_2)
def test__ne__(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.__ne__`
method.
"""
multi_signals_1 = self._multi_signals.copy()
multi_signals_2 = self._multi_signals.copy()
multi_signals_2[0] = 20
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2[0] = np.array([10, 20, 30])
self.assertEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator = CubicSplineInterpolator
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator = KernelInterpolator
self.assertEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator_args = {'window': 1}
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator_args = {}
self.assertEqual(multi_signals_1, multi_signals_2)
class NotExtrapolator(Extrapolator):
"""
Not :class:`Extrapolator` class.
"""
pass
multi_signals_2.extrapolator = NotExtrapolator
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.extrapolator = Extrapolator
self.assertEqual(multi_signals_1, multi_signals_2)
multi_signals_2.extrapolator_args = {}
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.extrapolator_args = {
'method': 'Constant',
'left': np.nan,
'right': np.nan
}
self.assertEqual(multi_signals_1, multi_signals_2)
def test_arithmetical_operation(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.\
arithmetical_operation` method.
"""
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, '+', False).range,
self._range_2 + 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, '-', False).range,
self._range_2 - 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, '*', False).range,
self._range_2 * 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, '/', False).range,
self._range_2 / 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, '**', False).range,
self._range_2**10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals + 10).range,
self._range_2 + 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals - 10).range,
self._range_2 - 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals * 10).range,
self._range_2 * 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals / 10).range,
self._range_2 / 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals**10).range,
self._range_2**10,
decimal=7)
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(multi_signals.arithmetical_operation(
10, '+', True).range,
self._range_2 + 10,
decimal=7)
np.testing.assert_almost_equal(multi_signals.arithmetical_operation(
10, '-', True).range,
self._range_2,
decimal=7)
np.testing.assert_almost_equal(multi_signals.arithmetical_operation(
10, '*', True).range,
self._range_2 * 10,
decimal=7)
np.testing.assert_almost_equal(multi_signals.arithmetical_operation(
10, '/', True).range,
self._range_2,
decimal=7)
np.testing.assert_almost_equal(multi_signals.arithmetical_operation(
10, '**', True).range,
self._range_2**10,
decimal=7)
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(multi_signals.arithmetical_operation(
self._range_2, '+', False).range,
self._range_2 + self._range_2,
decimal=7)
np.testing.assert_almost_equal(multi_signals.arithmetical_operation(
multi_signals, '+', False).range,
self._range_2 + self._range_2,
decimal=7)
def test_is_uniform(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.is_uniform`
method.
"""
self.assertTrue(self._multi_signals.is_uniform())
multi_signals = self._multi_signals.copy()
multi_signals[0.5] = 1.0
self.assertFalse(multi_signals.is_uniform())
def test_copy(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.copy` method.
"""
self.assertIsNot(self._multi_signals, self._multi_signals.copy())
self.assertEqual(self._multi_signals, self._multi_signals.copy())
def test_multi_signals_unpack_data(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.\
multi_signals_unpack_data` method.
"""
signals = MultiSignals.multi_signals_unpack_data(self._range_1)
self.assertListEqual(list(signals.keys()), [0])
np.testing.assert_array_equal(signals[0].domain, self._domain_1)
np.testing.assert_array_equal(signals[0].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_1, self._domain_2)
self.assertListEqual(list(signals.keys()), [0])
np.testing.assert_array_equal(signals[0].domain, self._domain_2)
np.testing.assert_array_equal(signals[0].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_2, self._domain_2)
self.assertListEqual(list(signals.keys()), [0, 1, 2])
np.testing.assert_array_equal(signals[0].range, self._range_1)
np.testing.assert_array_equal(signals[1].range, self._range_1 + 10)
np.testing.assert_array_equal(signals[2].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2)))
self.assertListEqual(list(signals.keys()), [0, 1, 2])
np.testing.assert_array_equal(signals[0].range, self._range_1)
np.testing.assert_array_equal(signals[1].range, self._range_1 + 10)
np.testing.assert_array_equal(signals[2].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))))
self.assertListEqual(list(signals.keys()), [0, 1, 2])
np.testing.assert_array_equal(signals[0].range, self._range_1)
np.testing.assert_array_equal(signals[1].range, self._range_1 + 10)
np.testing.assert_array_equal(signals[2].range, self._range_1 + 20)
if is_pandas_installed():
from pandas import DataFrame, Series
signals = MultiSignals.multi_signals_unpack_data(
Series(dict(zip(self._domain_1, self._range_1))))
self.assertListEqual(list(signals.keys()), [0])
np.testing.assert_array_equal(signals[0].domain, self._domain_1)
np.testing.assert_array_equal(signals[0].range, self._range_1)
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
signals = MultiSignals.multi_signals_unpack_data(
DataFrame(data, self._domain_1))
self.assertListEqual(list(signals.keys()), ['a', 'b', 'c'])
np.testing.assert_array_equal(signals['a'].range, self._range_1)
np.testing.assert_array_equal(signals['b'].range,
self._range_1 + 10)
np.testing.assert_array_equal(signals['c'].range,
self._range_1 + 20)
def test_fill_nan(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.fill_nan`
method.
"""
multi_signals = self._multi_signals.copy()
multi_signals[3:7] = np.nan
np.testing.assert_almost_equal(multi_signals.fill_nan().range,
np.array([[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 60.0, 70.0],
[60.0, 70.0, 80.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0]]),
decimal=7)
multi_signals[3:7] = np.nan
np.testing.assert_almost_equal(
multi_signals.fill_nan(method='Constant').range,
np.array([[10.0, 20.0, 30.0], [20.0, 30.0, 40.0],
[30.0, 40.0, 50.0], [0.0, 0.0, 0.0], [0.0, 0.0, 0.0],
[0.0, 0.0, 0.0], [0.0, 0.0, 0.0], [80.0, 90.0, 100.0],
[90.0, 100.0, 110.0], [100.0, 110.0, 120.0]]),
decimal=7)
def test_domain_distance(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.\
domain_distance` method.
"""
self.assertAlmostEqual(self._multi_signals.domain_distance(0.5),
0.5,
places=7)
np.testing.assert_almost_equal(
self._multi_signals.domain_distance(np.linspace(0, 9, 10) + 0.5),
np.array([0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5]),
decimal=7)
def test_to_dataframe(self):
"""
Tests :func:`colour.continuous.multi_signals.MultiSignals.to_dataframe`
method.
"""
if is_pandas_installed():
from pandas import DataFrame
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
assert MultiSignals(self._range_2,
self._domain_2,
labels=['a', 'b', 'c']).to_dataframe().equals(
DataFrame(data, self._domain_2))
def test_multi_signals_unpack_data(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.\
multi_signals_unpack_data` method.
"""
signals = MultiSignals.multi_signals_unpack_data(self._range_1)
self.assertListEqual(list(signals.keys()), ["0"])
np.testing.assert_array_equal(signals["0"].domain, self._domain_1)
np.testing.assert_array_equal(signals["0"].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_1, self._domain_2)
self.assertListEqual(list(signals.keys()), ["0"])
np.testing.assert_array_equal(signals["0"].domain, self._domain_2)
np.testing.assert_array_equal(signals["0"].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_1,
dict(zip(self._domain_2, self._range_1)).keys())
np.testing.assert_array_equal(signals["0"].domain, self._domain_2)
signals = MultiSignals.multi_signals_unpack_data(
self._range_2, self._domain_2)
self.assertListEqual(list(signals.keys()), ["0", "1", "2"])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
list(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))).values())[0])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))).values())
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2)))
self.assertListEqual(list(signals.keys()), ["0", "1", "2"])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))))
self.assertListEqual(list(signals.keys()), ["0", "1", "2"])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
if is_pandas_installed():
from pandas import DataFrame, Series
signals = MultiSignals.multi_signals_unpack_data(
Series(dict(zip(self._domain_1, self._range_1))))
self.assertListEqual(list(signals.keys()), ["0"])
np.testing.assert_array_equal(signals["0"].domain, self._domain_1)
np.testing.assert_array_equal(signals["0"].range, self._range_1)
data = dict(zip(["a", "b", "c"], tsplit(self._range_2)))
signals = MultiSignals.multi_signals_unpack_data(
DataFrame(data, self._domain_1))
self.assertListEqual(list(signals.keys()), ["a", "b", "c"])
np.testing.assert_array_equal(signals["a"].range, self._range_1)
np.testing.assert_array_equal(signals["b"].range,
self._range_1 + 10)
np.testing.assert_array_equal(signals["c"].range,
self._range_1 + 20)
class TestMultiSignals(unittest.TestCase):
"""
Define :class:`colour.continuous.multi_signals.MultiSignals` class unit
tests methods.
"""
def setUp(self):
"""Initialise the common tests attributes."""
self._range_1 = np.linspace(10, 100, 10)
self._range_2 = tstack([self._range_1] * 3) + np.array([0, 10, 20])
self._domain_1 = np.arange(0, 10, 1)
self._domain_2 = np.arange(100, 1100, 100)
self._multi_signals = MultiSignals(self._range_2)
def test_required_attributes(self):
"""Test the presence of required attributes."""
required_attributes = (
"dtype",
"domain",
"range",
"interpolator",
"interpolator_kwargs",
"extrapolator",
"extrapolator_kwargs",
"function",
"signals",
"labels",
"signal_type",
)
for attribute in required_attributes:
self.assertIn(attribute, dir(MultiSignals))
def test_required_methods(self):
"""Test the presence of required methods."""
required_methods = (
"__init__",
"__str__",
"__repr__",
"__hash__",
"__getitem__",
"__setitem__",
"__contains__",
"__eq__",
"__ne__",
"arithmetical_operation",
"multi_signals_unpack_data",
"fill_nan",
"domain_distance",
"to_dataframe",
)
for method in required_methods:
self.assertIn(method, dir(MultiSignals))
def test_dtype(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.dtype`
property.
"""
self.assertEqual(self._multi_signals.dtype, DEFAULT_FLOAT_DTYPE)
multi_signals = self._multi_signals.copy()
multi_signals.dtype = np.float32
self.assertEqual(multi_signals.dtype, np.float32)
def test_domain(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.domain`
property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 20.0, 30.0], [20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]),
decimal=7,
)
multi_signals.domain = self._domain_1 * 10
np.testing.assert_array_equal(multi_signals.domain,
self._domain_1 * 10)
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2]) * 10],
np.array([[10.0, 20.0, 30.0], [20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]),
decimal=7,
)
domain = np.linspace(0, 1, 10)
domain[0] = -np.inf
def assert_warns():
"""Help to test the runtime warning."""
multi_signals.domain = domain
self.assertWarns(ColourRuntimeWarning, assert_warns)
def test_range(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.range`
property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 20.0, 30.0], [20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]),
decimal=7,
)
multi_signals.range = self._range_1 * 10
np.testing.assert_array_equal(multi_signals.range,
tstack([self._range_1] * 3) * 10)
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 10.0, 10.0], [20.0, 20.0, 20.0],
[30.0, 30.0, 30.0]]) * 10,
decimal=7,
)
multi_signals.range = self._range_2 * 10
np.testing.assert_array_equal(multi_signals.range, self._range_2 * 10)
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2])],
np.array([[10.0, 20.0, 30.0], [20.0, 30.0, 40.0],
[30.0, 40.0, 50.0]]) * 10,
decimal=7,
)
def test_interpolator(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.interpolator`
property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([
[10.00000000, 20.00000000, 30.00000000],
[22.83489024, 32.80460562, 42.77432100],
[34.80044921, 44.74343470, 54.68642018],
[47.55353925, 57.52325463, 67.49297001],
[60.00000000, 70.00000000, 80.00000000],
]),
decimal=7,
)
multi_signals.interpolator = CubicSplineInterpolator
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([
[10.00000000, 20.00000000, 30.00000000],
[22.50000000, 32.50000000, 42.50000000],
[35.00000000, 45.00000000, 55.00000000],
[47.50000000, 57.50000000, 67.50000000],
[60.00000000, 70.00000000, 80.00000000],
]),
decimal=7,
)
def test_interpolator_kwargs(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.\
interpolator_kwargs` property.
"""
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([
[10.00000000, 20.00000000, 30.00000000],
[22.83489024, 32.80460562, 42.77432100],
[34.80044921, 44.74343470, 54.68642018],
[47.55353925, 57.52325463, 67.49297001],
[60.00000000, 70.00000000, 80.00000000],
]),
decimal=7,
)
multi_signals.interpolator_kwargs = {
"window": 1,
"kernel_kwargs": {
"a": 1
},
}
np.testing.assert_almost_equal(
multi_signals[np.linspace(0, 5, 5)],
np.array([
[10.00000000, 20.00000000, 30.00000000],
[18.91328761, 27.91961505, 36.92594248],
[28.36993142, 36.47562611, 44.58132080],
[44.13100443, 53.13733187, 62.14365930],
[60.00000000, 70.00000000, 80.00000000],
]),
decimal=7,
)
def test_extrapolator(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.extrapolator`
property.
"""
self.assertIsInstance(self._multi_signals.extrapolator(), Extrapolator)
def test_extrapolator_kwargs(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.\
extrapolator_kwargs` property.
"""
multi_signals = self._multi_signals.copy()
attest(np.all(np.isnan(multi_signals[np.array([-1000, 1000])])))
multi_signals.extrapolator_kwargs = {
"method": "Linear",
}
np.testing.assert_almost_equal(
multi_signals[np.array([-1000, 1000])],
np.array([[-9990.0, -9980.0, -9970.0], [10010.0, 10020.0,
10030.0]]),
decimal=7,
)
def test_function(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.function`
property.
"""
attest(hasattr(self._multi_signals.function, "__call__"))
def test_raise_exception_function(self):
"""
Test :func:`colour.continuous.signal.multi_signals.MultiSignals.\
function` property raised exception.
"""
self.assertRaises((ValueError, TypeError), MultiSignals().function, 0)
def test_signals(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.signals`
property.
"""
multi_signals = self._multi_signals.copy()
multi_signals.signals = self._range_1
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
def test_labels(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.labels`
property.
"""
self.assertListEqual(self._multi_signals.labels, ["0", "1", "2"])
multi_signals = self._multi_signals.copy()
multi_signals.labels = ["a", "b", "c"]
self.assertListEqual(multi_signals.labels, ["a", "b", "c"])
def test_signal_type(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.signal_type`
property.
"""
multi_signals = MultiSignals(signal_type=Signal)
self.assertEqual(multi_signals.signal_type, Signal)
def test__init__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__init__`
method.
"""
multi_signals = MultiSignals(self._range_1)
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
multi_signals = MultiSignals(self._range_1, self._domain_2)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
multi_signals = MultiSignals(self._range_2, self._domain_2)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
multi_signals = MultiSignals(dict(zip(self._domain_2, self._range_2)))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
multi_signals = MultiSignals(multi_signals)
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
class NotSignal(Signal):
"""Not :class:`Signal` class."""
pass
multi_signals = MultiSignals(self._range_1, signal_type=NotSignal)
self.assertIsInstance(multi_signals.signals["0"], NotSignal)
np.testing.assert_array_equal(multi_signals.domain, self._domain_1)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
if is_pandas_installed():
from pandas import DataFrame, Series
multi_signals = MultiSignals(
Series(dict(zip(self._domain_2, self._range_1))))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range,
self._range_1[:, np.newaxis])
data = dict(zip(["a", "b", "c"], tsplit(self._range_2)))
multi_signals = MultiSignals(DataFrame(data, self._domain_2))
np.testing.assert_array_equal(multi_signals.domain, self._domain_2)
np.testing.assert_array_equal(multi_signals.range, self._range_2)
def test__hash__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__hash__`
method.
"""
self.assertIsInstance(hash(self._multi_signals), int)
def test__str__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__str__`
method.
"""
self.assertEqual(
str(self._multi_signals),
textwrap.dedent("""
[[ 0. 10. 20. 30.]
[ 1. 20. 30. 40.]
[ 2. 30. 40. 50.]
[ 3. 40. 50. 60.]
[ 4. 50. 60. 70.]
[ 5. 60. 70. 80.]
[ 6. 70. 80. 90.]
[ 7. 80. 90. 100.]
[ 8. 90. 100. 110.]
[ 9. 100. 110. 120.]]""")[1:],
)
self.assertIsInstance(str(MultiSignals()), str)
def test__repr__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__repr__`
method.
"""
self.assertEqual(
re.sub(
r"extrapolator_kwargs={.*}",
"extrapolator_kwargs={...}",
repr(self._multi_signals),
),
textwrap.dedent("""
MultiSignals([[ 0., 10., 20., 30.],
[ 1., 20., 30., 40.],
[ 2., 30., 40., 50.],
[ 3., 40., 50., 60.],
[ 4., 50., 60., 70.],
[ 5., 60., 70., 80.],
[ 6., 70., 80., 90.],
[ 7., 80., 90., 100.],
[ 8., 90., 100., 110.],
[ 9., 100., 110., 120.]],
labels=['0', '1', '2'],
interpolator=KernelInterpolator,
interpolator_kwargs={},
extrapolator=Extrapolator,
extrapolator_kwargs={...})""")[1:],
)
self.assertIsInstance(repr(MultiSignals()), str)
def test__getitem__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__getitem__`
method.
"""
np.testing.assert_almost_equal(self._multi_signals[0],
np.array([10.0, 20.0, 30.0]),
decimal=7)
np.testing.assert_almost_equal(
self._multi_signals[np.array([0, 1, 2])],
np.array([
[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0],
]),
decimal=7,
)
np.testing.assert_almost_equal(
self._multi_signals[np.linspace(0, 5, 5)],
np.array([
[10.00000000, 20.00000000, 30.00000000],
[22.83489024, 32.80460562, 42.77432100],
[34.80044921, 44.74343470, 54.68642018],
[47.55353925, 57.52325463, 67.49297001],
[60.00000000, 70.00000000, 80.00000000],
]),
decimal=7,
)
attest(np.all(np.isnan(self._multi_signals[np.array([-1000, 1000])])))
np.testing.assert_almost_equal(self._multi_signals[:],
self._multi_signals.range,
decimal=7)
np.testing.assert_almost_equal(self._multi_signals[:, :],
self._multi_signals.range,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signals[0:3],
np.array([
[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0],
]),
decimal=7,
)
np.testing.assert_almost_equal(
self._multi_signals[:, 0:2],
np.array([
[10.0, 20.0],
[20.0, 30.0],
[30.0, 40.0],
[40.0, 50.0],
[50.0, 60.0],
[60.0, 70.0],
[70.0, 80.0],
[80.0, 90.0],
[90.0, 100.0],
[100.0, 110.0],
]),
decimal=7,
)
multi_signals = self._multi_signals.copy()
multi_signals.extrapolator_kwargs = {
"method": "Linear",
}
np.testing.assert_array_equal(
multi_signals[np.array([-1000, 1000])],
np.array([
[-9990.0, -9980.0, -9970.0],
[10010.0, 10020.0, 10030.0],
]),
)
multi_signals.extrapolator_kwargs = {
"method": "Constant",
"left": 0,
"right": 1,
}
np.testing.assert_array_equal(
multi_signals[np.array([-1000, 1000])],
np.array([[0.0, 0.0, 0.0], [1.0, 1.0, 1.0]]),
)
def test__setitem__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__setitem__`
method.
"""
multi_signals = self._multi_signals.copy()
multi_signals[0] = 20
np.testing.assert_almost_equal(multi_signals[0],
np.array([20.0, 20.0, 20.0]),
decimal=7)
multi_signals[np.array([0, 1, 2])] = 30
np.testing.assert_almost_equal(
multi_signals[np.array([0, 1, 2])],
np.array([
[30.0, 30.0, 30.0],
[30.0, 30.0, 30.0],
[30.0, 30.0, 30.0],
]),
decimal=7,
)
multi_signals[np.linspace(0, 5, 5)] = 50
np.testing.assert_almost_equal(
multi_signals.domain,
np.array([
0.00,
1.00,
1.25,
2.00,
2.50,
3.00,
3.75,
4.00,
5.00,
6.00,
7.00,
8.00,
9.00,
]),
decimal=7,
)
np.testing.assert_almost_equal(
multi_signals.range,
np.array([
[50.0, 50.0, 50.0],
[30.0, 30.0, 30.0],
[50.0, 50.0, 50.0],
[30.0, 30.0, 30.0],
[50.0, 50.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 50.0, 50.0],
[50.0, 60.0, 70.0],
[50.0, 50.0, 50.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7,
)
multi_signals[np.array([0, 1, 2])] = np.array([10, 20, 30])
np.testing.assert_almost_equal(
multi_signals.range,
np.array([
[10.0, 20.0, 30.0],
[10.0, 20.0, 30.0],
[50.0, 50.0, 50.0],
[10.0, 20.0, 30.0],
[50.0, 50.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 50.0, 50.0],
[50.0, 60.0, 70.0],
[50.0, 50.0, 50.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7,
)
multi_signals[np.array([0, 1,
2])] = np.reshape(np.arange(1, 10, 1), (3, 3))
np.testing.assert_almost_equal(
multi_signals.range,
np.array([
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
[50.0, 50.0, 50.0],
[7.0, 8.0, 9.0],
[50.0, 50.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 50.0, 50.0],
[50.0, 60.0, 70.0],
[50.0, 50.0, 50.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7,
)
multi_signals[:] = 40
np.testing.assert_almost_equal(multi_signals.range, 40, decimal=7)
multi_signals[:, :] = 50
np.testing.assert_almost_equal(multi_signals.range, 50, decimal=7)
multi_signals = self._multi_signals.copy()
multi_signals[0:3] = 40
np.testing.assert_almost_equal(
multi_signals[0:3],
np.array([
[40.0, 40.0, 40.0],
[40.0, 40.0, 40.0],
[40.0, 40.0, 40.0],
]),
decimal=7,
)
multi_signals[:, 0:2] = 50
np.testing.assert_almost_equal(
multi_signals.range,
np.array([
[50.0, 50.0, 40.0],
[50.0, 50.0, 40.0],
[50.0, 50.0, 40.0],
[50.0, 50.0, 60.0],
[50.0, 50.0, 70.0],
[50.0, 50.0, 80.0],
[50.0, 50.0, 90.0],
[50.0, 50.0, 100.0],
[50.0, 50.0, 110.0],
[50.0, 50.0, 120.0],
]),
decimal=7,
)
def test__contains__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__contains__`
method.
"""
self.assertIn(0, self._multi_signals)
self.assertIn(0.5, self._multi_signals)
self.assertNotIn(1000, self._multi_signals)
def test__len__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__len__`
method.
"""
self.assertEqual(len(self._multi_signals), 10)
def test__eq__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__eq__`
method.
"""
signal_1 = self._multi_signals.copy()
signal_2 = self._multi_signals.copy()
self.assertEqual(signal_1, signal_2)
self.assertNotEqual(signal_1, None)
def test__ne__(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.__ne__`
method.
"""
multi_signals_1 = self._multi_signals.copy()
multi_signals_2 = self._multi_signals.copy()
multi_signals_2[0] = 20
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2[0] = np.array([10, 20, 30])
self.assertEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator = CubicSplineInterpolator
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator = KernelInterpolator
self.assertEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator_kwargs = {"window": 1}
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.interpolator_kwargs = {}
self.assertEqual(multi_signals_1, multi_signals_2)
class NotExtrapolator(Extrapolator):
"""Not :class:`Extrapolator` class."""
pass
multi_signals_2.extrapolator = NotExtrapolator
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.extrapolator = Extrapolator
self.assertEqual(multi_signals_1, multi_signals_2)
multi_signals_2.extrapolator_kwargs = {}
self.assertNotEqual(multi_signals_1, multi_signals_2)
multi_signals_2.extrapolator_kwargs = {
"method": "Constant",
"left": np.nan,
"right": np.nan,
}
self.assertEqual(multi_signals_1, multi_signals_2)
def test_arithmetical_operation(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.\
arithmetical_operation` method.
"""
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, "+", False).range,
self._range_2 + 10,
decimal=7,
)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, "-", False).range,
self._range_2 - 10,
decimal=7,
)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, "*", False).range,
self._range_2 * 10,
decimal=7,
)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, "/", False).range,
self._range_2 / 10,
decimal=7,
)
np.testing.assert_almost_equal(
self._multi_signals.arithmetical_operation(10, "**", False).range,
self._range_2**10,
decimal=7,
)
np.testing.assert_almost_equal((self._multi_signals + 10).range,
self._range_2 + 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals - 10).range,
self._range_2 - 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals * 10).range,
self._range_2 * 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals / 10).range,
self._range_2 / 10,
decimal=7)
np.testing.assert_almost_equal((self._multi_signals**10).range,
self._range_2**10,
decimal=7)
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals.arithmetical_operation(10, "+", True).range,
self._range_2 + 10,
decimal=7,
)
np.testing.assert_almost_equal(
multi_signals.arithmetical_operation(10, "-", True).range,
self._range_2,
decimal=7,
)
np.testing.assert_almost_equal(
multi_signals.arithmetical_operation(10, "*", True).range,
self._range_2 * 10,
decimal=7,
)
np.testing.assert_almost_equal(
multi_signals.arithmetical_operation(10, "/", True).range,
self._range_2,
decimal=7,
)
np.testing.assert_almost_equal(
multi_signals.arithmetical_operation(10, "**", True).range,
self._range_2**10,
decimal=7,
)
multi_signals = self._multi_signals.copy()
np.testing.assert_almost_equal(
multi_signals.arithmetical_operation(self._range_2, "+",
False).range,
self._range_2 + self._range_2,
decimal=7,
)
np.testing.assert_almost_equal(
multi_signals.arithmetical_operation(multi_signals, "+",
False).range,
self._range_2 + self._range_2,
decimal=7,
)
def test_is_uniform(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.is_uniform`
method.
"""
self.assertTrue(self._multi_signals.is_uniform())
multi_signals = self._multi_signals.copy()
multi_signals[0.5] = 1.0
self.assertFalse(multi_signals.is_uniform())
def test_copy(self):
"""Test :func:`colour.continuous.multi_signals.MultiSignals.copy` method."""
self.assertIsNot(self._multi_signals, self._multi_signals.copy())
self.assertEqual(self._multi_signals, self._multi_signals.copy())
def test_multi_signals_unpack_data(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.\
multi_signals_unpack_data` method.
"""
signals = MultiSignals.multi_signals_unpack_data(self._range_1)
self.assertListEqual(list(signals.keys()), ["0"])
np.testing.assert_array_equal(signals["0"].domain, self._domain_1)
np.testing.assert_array_equal(signals["0"].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_1, self._domain_2)
self.assertListEqual(list(signals.keys()), ["0"])
np.testing.assert_array_equal(signals["0"].domain, self._domain_2)
np.testing.assert_array_equal(signals["0"].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
self._range_1,
dict(zip(self._domain_2, self._range_1)).keys())
np.testing.assert_array_equal(signals["0"].domain, self._domain_2)
signals = MultiSignals.multi_signals_unpack_data(
self._range_2, self._domain_2)
self.assertListEqual(list(signals.keys()), ["0", "1", "2"])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
list(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))).values())[0])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
signals = MultiSignals.multi_signals_unpack_data(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))).values())
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2)))
self.assertListEqual(list(signals.keys()), ["0", "1", "2"])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
signals = MultiSignals.multi_signals_unpack_data(
MultiSignals.multi_signals_unpack_data(
dict(zip(self._domain_2, self._range_2))))
self.assertListEqual(list(signals.keys()), ["0", "1", "2"])
np.testing.assert_array_equal(signals["0"].range, self._range_1)
np.testing.assert_array_equal(signals["1"].range, self._range_1 + 10)
np.testing.assert_array_equal(signals["2"].range, self._range_1 + 20)
if is_pandas_installed():
from pandas import DataFrame, Series
signals = MultiSignals.multi_signals_unpack_data(
Series(dict(zip(self._domain_1, self._range_1))))
self.assertListEqual(list(signals.keys()), ["0"])
np.testing.assert_array_equal(signals["0"].domain, self._domain_1)
np.testing.assert_array_equal(signals["0"].range, self._range_1)
data = dict(zip(["a", "b", "c"], tsplit(self._range_2)))
signals = MultiSignals.multi_signals_unpack_data(
DataFrame(data, self._domain_1))
self.assertListEqual(list(signals.keys()), ["a", "b", "c"])
np.testing.assert_array_equal(signals["a"].range, self._range_1)
np.testing.assert_array_equal(signals["b"].range,
self._range_1 + 10)
np.testing.assert_array_equal(signals["c"].range,
self._range_1 + 20)
def test_fill_nan(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.fill_nan`
method.
"""
multi_signals = self._multi_signals.copy()
multi_signals[3:7] = np.nan
np.testing.assert_almost_equal(
multi_signals.fill_nan().range,
np.array([
[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0],
[40.0, 50.0, 60.0],
[50.0, 60.0, 70.0],
[60.0, 70.0, 80.0],
[70.0, 80.0, 90.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7,
)
multi_signals[3:7] = np.nan
np.testing.assert_almost_equal(
multi_signals.fill_nan(method="Constant").range,
np.array([
[10.0, 20.0, 30.0],
[20.0, 30.0, 40.0],
[30.0, 40.0, 50.0],
[0.0, 0.0, 0.0],
[0.0, 0.0, 0.0],
[0.0, 0.0, 0.0],
[0.0, 0.0, 0.0],
[80.0, 90.0, 100.0],
[90.0, 100.0, 110.0],
[100.0, 110.0, 120.0],
]),
decimal=7,
)
def test_domain_distance(self):
"""
Test :func:`colour.continuous.multi_signals.MultiSignals.\
domain_distance` method.
"""
self.assertAlmostEqual(self._multi_signals.domain_distance(0.5),
0.5,
places=7)
np.testing.assert_almost_equal(
self._multi_signals.domain_distance(np.linspace(0, 9, 10) + 0.5),
np.array([0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5]),
decimal=7,
)
def test_to_dataframe(self):
"""
Test :meth:`colour.continuous.multi_signals.MultiSignals.to_dataframe`
method.
"""
if is_pandas_installed():
from pandas import DataFrame
data = dict(zip(["a", "b", "c"], tsplit(self._range_2)))
attest(
MultiSignals(self._range_2,
self._domain_2,
labels=["a", "b", "c"]).to_dataframe().equals(
DataFrame(data, self._domain_2)))
