testing

Author: roytsmart

named_arrays/_functions/functions.py

@@ -1017,7 +1017,7 @@ def __setitem__(
                     item_inputs[ax] = item_outputs[ax] = item_ax
         else:
             raise TypeError(
-                f"`item` must be an instance of `{dict.__name__}`, or `{na.AbstractFunctionArray.__name__}`, "
+                f"`item` must be an instance of `bool`, `{dict.__name__}`, or `{na.AbstractFunctionArray.__name__}`, "
                 f"got `{type(item)}`"
             )
 

named_arrays/_functions/tests/test_functions.py

@@ -194,6 +194,7 @@ def test_interp_linear_identity(
 
         with pytest.raises(NotImplementedError):
             array.interp_linear(array.indices)
+
     @pytest.mark.parametrize(
         argnames='item',
         argvalues=[
@@ -206,6 +207,7 @@ def test_interp_linear_identity(
                     outputs=na.ScalarArrayRange(0, 2, axis='y'),
                 )
             ),
+            True,
             na.ScalarLinearSpace(0, 1, axis='y', num=_num_y) > 0.5,
             na.FunctionArray(
                 inputs=na.ScalarLinearSpace(0, 1, axis='y', num=_num_y),
@@ -225,9 +227,6 @@ def test_interp_linear_identity(
             )
         ],
     )
-
-
-
     def test__getitem__(
             self,
             array: na.AbstractFunctionArray,
@@ -947,6 +946,7 @@ class TestFunctionArray(
             dict(y=slice(None)),
             dict(y=na.ScalarArrayRange(0, _num_y, axis='y')),
             dict(x=na.ScalarArrayRange(0, _num_x, axis='x'), y=na.ScalarArrayRange(0, _num_y, axis='y')),
+            True,
             na.FunctionArray(
                 inputs=na.ScalarLinearSpace(0, 1, axis='y', num=_num_y),
                 outputs=na.ScalarArray.ones(shape=dict(y=_num_y), dtype=bool),

named_arrays/_scalars/scalars.py

@@ -1086,6 +1086,12 @@ def __setitem__(
 
             self.ndarray_aligned(axes_self)[tuple(index)] = value
 
+        else:
+            raise TypeError(
+                f"`item` must be an instance of `bool`, `{na.AbstractArray.__name__}`, or {dict.__name__}, "
+                f"got `{type(item)}`"
+            )
+
 
 @dataclasses.dataclass(eq=False, repr=False)
 class AbstractImplicitScalarArray(

named_arrays/_scalars/tests/test_scalars.py

@@ -425,6 +425,7 @@ def test_change_axis_index(self, array: na.ScalarArray, index: int):
             dict(y=0),
             dict(y=slice(0,1)),
             dict(y=na.ScalarArray(np.array([0, 1]), axes=('y', ))),
+            True,
             na.ScalarLinearSpace(0, 1, axis='y', num=_num_y) > 0.5,
         ]
     )
@@ -1319,6 +1320,7 @@ class TestScalarArray(
             dict(y=slice(None)),
             dict(y=na.ScalarArrayRange(0, _num_y, axis='y')),
             dict(x=na.ScalarArrayRange(0, _num_x, axis='x'), y=na.ScalarArrayRange(0, _num_y, axis='y')),
+            True,
             na.ScalarArray.ones(shape=dict(y=_num_y), dtype=bool),
         ],
     )

named_arrays/_scalars/uncertainties/tests/test_uncertainties.py

@@ -114,6 +114,7 @@ def test_shape_distribution(self, array: na.AbstractUncertainScalarArray):
                     )
                 )
             ),
+            True,
             na.ScalarLinearSpace(0, 1, axis='y', num=_num_y) > 0.5,
             na.UncertainScalarArray(
                 nominal=na.ScalarLinearSpace(0, 1, axis='y', num=_num_y),
@@ -795,6 +796,7 @@ class TestUncertainScalarArray(
             dict(y=slice(None)),
             dict(y=na.ScalarArrayRange(0, _num_y, axis='y')),
             dict(x=na.ScalarArrayRange(0, _num_x, axis='x'), y=na.ScalarArrayRange(0, _num_y, axis='y')),
+            True,
             na.ScalarArray.ones(shape=dict(y=_num_y), dtype=bool),
         ],
     )

named_arrays/_scalars/uncertainties/uncertainties.py

@@ -703,7 +703,7 @@ def __setitem__(
 
         else:
             raise TypeError(
-                f"`item` must be an instance of `{na.AbstractArray.__name__}` or {dict.__name__}, "
+                f"`item` must be an instance of `bool`, `{na.AbstractArray.__name__}`, or {dict.__name__}, "
                 f"got `{type(item)}`"
             )
 

named_arrays/_vectors/cartesian/tests/test_vectors_cartesian_2d.py

@@ -108,6 +108,7 @@ def _cartesian2d_items() -> list[na.AbstractArray | dict[str, int, slice, na.Abs
                     )
                 )
             ),
+            True,
             na.ScalarLinearSpace(0, 1, axis='y', num=_num_y) > 0.5,
             na.UniformUncertainScalarArray(
                 nominal=na.ScalarLinearSpace(0, 1, axis='y', num=_num_y),
@@ -259,6 +260,7 @@ class TestCartesian2dVectorArray(
                     y=na.ScalarArrayRange(0, _num_y, axis='y'),
                 )
             ),
+            True,
             na.ScalarArray.ones(shape=dict(y=_num_y), dtype=bool),
             np.ones_like(na.Cartesian2dVectorArray(), dtype=bool, shape=dict(y=_num_y)),
         ],

named_arrays/_vectors/vectors.py

@@ -668,7 +668,7 @@ def __setitem__(
 
         else:
             raise TypeError(
-                f"`item` must be an instance of `{na.AbstractArray.__name__}` or {dict.__name__}, "
+                f"`item` must be an instance of `bool`, `{na.AbstractArray.__name__}`, or {dict.__name__}, "
                 f"got `{type(item)}`"
             )