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_signal = MultiSignal(self._range_2)
def test_multi_signal_unpack_data(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.\
multi_signal_unpack_data` method.
"""
signals = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_unpack_data(
MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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_signal.MultiSignal.signal_type`
property.
"""
multi_signal = MultiSignal(signal_type=Signal)
self.assertEqual(multi_signal.signal_type, Signal)
def test__init__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__init__`
method.
"""
multi_signal = MultiSignal(self._range_1)
np.testing.assert_array_equal(multi_signal.domain, self._domain_1)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
multi_signal = MultiSignal(self._range_1, self._domain_2)
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
multi_signal = MultiSignal(self._range_2, self._domain_2)
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
multi_signal = MultiSignal(dict(zip(self._domain_2, self._range_2)))
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
multi_signal = MultiSignal(multi_signal)
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
class NotSignal(Signal):
"""
Not :class:`Signal` class.
"""
pass
multi_signal = MultiSignal(self._range_1, signal_type=NotSignal)
self.assertIsInstance(multi_signal.signals[0], NotSignal)
np.testing.assert_array_equal(multi_signal.domain, self._domain_1)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
if is_pandas_installed():
from pandas import DataFrame, Series
multi_signal = MultiSignal(
Series(dict(zip(self._domain_2, self._range_1))))
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
multi_signal = MultiSignal(DataFrame(data, self._domain_2))
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
def test_to_dataframe(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.to_dataframe`
method.
"""
if is_pandas_installed():
from pandas import DataFrame
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
assert MultiSignal(
self._range_2, self._domain_2,
labels=['a', 'b', 'c']).to_dataframe().equals(
DataFrame(data, self._domain_2))
class TestMultiSignal(unittest.TestCase):
"""
Defines :class:`colour.continuous.multi_signal.MultiSignal` 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_signal = MultiSignal(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(MultiSignal))
def test_required_methods(self):
"""
Tests presence of required methods.
"""
required_methods = ('__str__', '__repr__', '__hash__', '__getitem__',
'__setitem__', '__contains__', '__eq__', '__ne__',
'arithmetical_operation',
'multi_signal_unpack_data', 'fill_nan',
'domain_distance', 'to_dataframe')
for method in required_methods:
self.assertIn(method, dir(MultiSignal))
def test_domain(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.domain`
property.
"""
multi_signal = self._multi_signal.copy()
np.testing.assert_almost_equal(
multi_signal[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_signal.domain = self._domain_1 * 10
np.testing.assert_array_equal(multi_signal.domain, self._domain_1 * 10)
np.testing.assert_almost_equal(
multi_signal[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)
def test_range(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.range`
property.
"""
multi_signal = self._multi_signal.copy()
np.testing.assert_almost_equal(
multi_signal[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_signal.range = self._range_1 * 10
np.testing.assert_array_equal(multi_signal.range,
tstack([self._range_1] * 3) * 10)
np.testing.assert_almost_equal(
multi_signal[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_signal.range = self._range_2 * 10
np.testing.assert_array_equal(multi_signal.range, self._range_2 * 10)
np.testing.assert_almost_equal(
multi_signal[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_signal.MultiSignal.interpolator`
property.
"""
multi_signal = self._multi_signal.copy()
np.testing.assert_almost_equal(
multi_signal[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_signal.interpolator = CubicSplineInterpolator
np.testing.assert_almost_equal(
multi_signal[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_signal.MultiSignal.\
interpolator_args` property.
"""
multi_signal = self._multi_signal.copy()
np.testing.assert_almost_equal(
multi_signal[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_signal.interpolator_args = {'window': 1, 'kernel_args': {'a': 1}}
np.testing.assert_almost_equal(
multi_signal[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_signal.MultiSignal.extrapolator`
property.
"""
self.assertIsInstance(self._multi_signal.extrapolator(), Extrapolator)
def test_extrapolator_args(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.\
extrapolator_args` property.
"""
multi_signal = self._multi_signal.copy()
assert np.all(np.isnan(multi_signal[np.array([-1000, 1000])]))
multi_signal.extrapolator_args = {
'method': 'Linear',
}
np.testing.assert_almost_equal(
multi_signal[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_signal.MultiSignal.function`
property.
"""
assert hasattr(self._multi_signal.function, '__call__')
def test_signals(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.signals`
property.
"""
multi_signal = self._multi_signal.copy()
multi_signal.signals = self._range_1
np.testing.assert_array_equal(multi_signal.domain, self._domain_1)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
def test_labels(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.labels`
property.
"""
self.assertListEqual(self._multi_signal.labels, [0, 1, 2])
multi_signal = self._multi_signal.copy()
multi_signal.labels = ['a', 'b', 'c']
self.assertListEqual(multi_signal.labels, ['a', 'b', 'c'])
def test_signal_type(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.signal_type`
property.
"""
multi_signal = MultiSignal(signal_type=Signal)
self.assertEqual(multi_signal.signal_type, Signal)
def test__init__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__init__`
method.
"""
multi_signal = MultiSignal(self._range_1)
np.testing.assert_array_equal(multi_signal.domain, self._domain_1)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
multi_signal = MultiSignal(self._range_1, self._domain_2)
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
multi_signal = MultiSignal(self._range_2, self._domain_2)
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
multi_signal = MultiSignal(dict(zip(self._domain_2, self._range_2)))
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
multi_signal = MultiSignal(multi_signal)
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
class NotSignal(Signal):
"""
Not :class:`Signal` class.
"""
pass
multi_signal = MultiSignal(self._range_1, signal_type=NotSignal)
self.assertIsInstance(multi_signal.signals[0], NotSignal)
np.testing.assert_array_equal(multi_signal.domain, self._domain_1)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
if is_pandas_installed():
from pandas import DataFrame, Series
multi_signal = MultiSignal(
Series(dict(zip(self._domain_2, self._range_1))))
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range,
self._range_1[:, np.newaxis])
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
multi_signal = MultiSignal(DataFrame(data, self._domain_2))
np.testing.assert_array_equal(multi_signal.domain, self._domain_2)
np.testing.assert_array_equal(multi_signal.range, self._range_2)
def test__str__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__str__`
method.
"""
self.assertEqual(
str(self._multi_signal),
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:])
def test__repr__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__repr__`
method.
"""
self.assertEqual(
re.sub(r'extrapolator_args={.*}', 'extrapolator_args={...}',
repr(self._multi_signal)),
textwrap.dedent("""
MultiSignal([[ 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:])
def test__getitem__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__getitem__`
method.
"""
np.testing.assert_almost_equal(
self._multi_signal[0], np.array([10.0, 20.0, 30.0]), decimal=7)
np.testing.assert_almost_equal(
self._multi_signal[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_signal[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_signal[np.array([-1000, 1000])]))
multi_signal = self._multi_signal.copy()
multi_signal.extrapolator_args = {
'method': 'Linear',
}
np.testing.assert_array_equal(
multi_signal[np.array([-1000, 1000])],
np.array([[-9990.0, -9980.0, -9970.0], [10010.0, 10020.0,
10030.0]]))
multi_signal.extrapolator_args = {
'method': 'Constant',
'left': 0,
'right': 1
}
np.testing.assert_array_equal(
multi_signal[np.array([-1000, 1000])],
np.array([[0.0, 0.0, 0.0], [1.0, 1.0, 1.0]]))
def test__setitem__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__setitem__`
method.
"""
multi_signal = self._multi_signal.copy()
multi_signal[0] = 20
np.testing.assert_almost_equal(
multi_signal[0], np.array([20.0, 20.0, 20.0]), decimal=7)
multi_signal[np.array([0, 1, 2])] = 30
np.testing.assert_almost_equal(
multi_signal[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_signal[0:3] = 40
np.testing.assert_almost_equal(
multi_signal[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_signal[np.linspace(0, 5, 5)] = 50
np.testing.assert_almost_equal(
multi_signal.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_signal.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_signal[np.array([0, 1, 2])] = np.array([10, 20, 30])
np.testing.assert_almost_equal(
multi_signal.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_signal[np.array([0, 1, 2])] = np.arange(1, 10, 1).reshape(3, 3)
np.testing.assert_almost_equal(
multi_signal.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_signal.MultiSignal.__contains__`
method.
"""
self.assertIn(0, self._multi_signal)
self.assertIn(0.5, self._multi_signal)
self.assertNotIn(1000, self._multi_signal)
def test__len__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__len__`
method.
"""
self.assertEqual(len(self._multi_signal), 10)
def test__eq__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__eq__` method.
"""
signal_1 = self._multi_signal.copy()
signal_2 = self._multi_signal.copy()
self.assertEqual(signal_1, signal_2)
def test__ne__(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.__ne__` method.
"""
multi_signal_1 = self._multi_signal.copy()
multi_signal_2 = self._multi_signal.copy()
multi_signal_2[0] = 20
self.assertNotEqual(multi_signal_1, multi_signal_2)
multi_signal_2[0] = np.array([10, 20, 30])
self.assertEqual(multi_signal_1, multi_signal_2)
multi_signal_2.interpolator = CubicSplineInterpolator
self.assertNotEqual(multi_signal_1, multi_signal_2)
multi_signal_2.interpolator = KernelInterpolator
self.assertEqual(multi_signal_1, multi_signal_2)
multi_signal_2.interpolator_args = {'window': 1}
self.assertNotEqual(multi_signal_1, multi_signal_2)
multi_signal_2.interpolator_args = {}
self.assertEqual(multi_signal_1, multi_signal_2)
class NotExtrapolator(Extrapolator):
"""
Not :class:`Extrapolator` class.
"""
pass
multi_signal_2.extrapolator = NotExtrapolator
self.assertNotEqual(multi_signal_1, multi_signal_2)
multi_signal_2.extrapolator = Extrapolator
self.assertEqual(multi_signal_1, multi_signal_2)
multi_signal_2.extrapolator_args = {}
self.assertNotEqual(multi_signal_1, multi_signal_2)
multi_signal_2.extrapolator_args = {
'method': 'Constant',
'left': np.nan,
'right': np.nan
}
self.assertEqual(multi_signal_1, multi_signal_2)
def test_arithmetical_operation(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.\
arithmetical_operation` method.
"""
np.testing.assert_almost_equal(
self._multi_signal.arithmetical_operation(10, '+', False).range,
self._range_2 + 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signal.arithmetical_operation(10, '-', False).range,
self._range_2 - 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signal.arithmetical_operation(10, '*', False).range,
self._range_2 * 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signal.arithmetical_operation(10, '/', False).range,
self._range_2 / 10,
decimal=7)
np.testing.assert_almost_equal(
self._multi_signal.arithmetical_operation(10, '**', False).range,
self._range_2 ** 10,
decimal=7)
multi_signal = self._multi_signal.copy()
np.testing.assert_almost_equal(
multi_signal.arithmetical_operation(10, '+', True).range,
self._range_2 + 10,
decimal=7)
np.testing.assert_almost_equal(
multi_signal.arithmetical_operation(10, '-', True).range,
self._range_2,
decimal=7)
np.testing.assert_almost_equal(
multi_signal.arithmetical_operation(10, '*', True).range,
self._range_2 * 10,
decimal=7)
np.testing.assert_almost_equal(
multi_signal.arithmetical_operation(10, '/', True).range,
self._range_2,
decimal=7)
np.testing.assert_almost_equal(
multi_signal.arithmetical_operation(10, '**', True).range,
self._range_2 ** 10,
decimal=7)
multi_signal = self._multi_signal.copy()
np.testing.assert_almost_equal(
multi_signal.arithmetical_operation(self._range_2, '+',
False).range,
self._range_2 + self._range_2,
decimal=7)
np.testing.assert_almost_equal(
multi_signal.arithmetical_operation(multi_signal, '+',
False).range,
self._range_2 + self._range_2,
decimal=7)
def test_is_uniform(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.is_uniform`
method.
"""
self.assertTrue(self._multi_signal.is_uniform())
multi_signal = self._multi_signal.copy()
multi_signal[0.5] = 1.0
self.assertFalse(multi_signal.is_uniform())
def test_copy(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.copy` method.
"""
self.assertIsNot(self._multi_signal, self._multi_signal.copy())
self.assertEqual(self._multi_signal, self._multi_signal.copy())
def test_multi_signal_unpack_data(self):
"""
Tests :func:`colour.continuous.multi_signal.MultiSignal.\
multi_signal_unpack_data` method.
"""
signals = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_unpack_data(
MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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 = MultiSignal.multi_signal_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_signal.MultiSignal.fill_nan`
method.
"""
multi_signal = self._multi_signal.copy()
multi_signal[3:7] = np.nan
np.testing.assert_almost_equal(
multi_signal.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_signal[3:7] = np.nan
np.testing.assert_almost_equal(
multi_signal.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_signal.MultiSignal.\
domain_distance` method.
"""
self.assertAlmostEqual(
self._multi_signal.domain_distance(0.5), 0.5, places=7)
np.testing.assert_almost_equal(
self._multi_signal.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_signal.MultiSignal.to_dataframe`
method.
"""
if is_pandas_installed():
from pandas import DataFrame
data = dict(zip(['a', 'b', 'c'], tsplit(self._range_2)))
assert MultiSignal(
self._range_2, self._domain_2,
labels=['a', 'b', 'c']).to_dataframe().equals(
DataFrame(data, self._domain_2))
