def test_n_dimensional_in_array(self): """ Tests :func:`colour.utilities.array.in_array` definition n-dimensional support. """ np.testing.assert_almost_equal( in_array(np.array([0.50, 0.60]), np.linspace(0, 10, 101)).shape, np.array([2])) np.testing.assert_almost_equal( in_array(np.array([[0.50, 0.60]]), np.linspace(0, 10, 101)).shape, np.array([1, 2])) np.testing.assert_almost_equal( in_array(np.array([[0.50], [0.60]]), np.linspace(0, 10, 101)).shape, np.array([2, 1]))
def test_in_array(self): """ Tests :func:`colour.utilities.array.in_array` definition. """ self.assertTrue( np.array_equal( in_array(np.array([0.50, 0.60]), np.linspace(0, 10, 101)), np.array([True, True]))) self.assertFalse( np.array_equal( in_array(np.array([0.50, 0.61]), np.linspace(0, 10, 101)), np.array([True, True]))) self.assertTrue( np.array_equal( in_array(np.array([[0.50], [0.60]]), np.linspace(0, 10, 101)), np.array([[True], [True]])))