Method inject_str

numpy/core/tests/test_multiarray.py:576–582

replace ndarray.__str__ temporarily

(s)

        @contextmanager
        def inject_str(s):
            """ replace ndarray.__str__ temporarily """
            set_string_function(lambda x: s, repr=False)
            try:
                yield
            finally:
                set_string_function(None, repr=False)

        a1d = np.array(['test'])
        a0d = np.array('done')