def test_out_shape_mismatch(self, func_name):
        """This test checks if we raise an error when the output shape is
        incorrect"""

        in_num = np.random.random((3, 3))
        func_num = getattr(num, func_name)

        # shape
        index_array = np.random.randint(low=1, high=4, size=(2, 2))
        func_num(in_num, out=index_array)
This test checks if we raise an error when the output shape is
        incorrect
tests/integration/test_nanarg_reduction.py