Method inject_str

numpy/_core/tests/test_multiarray.py:691–697

replace ndarray.__str__ temporarily

(s)

        @contextmanager
        def inject_str(s):
            """ replace ndarray.__str__ temporarily """
            set_printoptions(formatter={"all": lambda x: s})
            try:
                yield
            finally:
                set_printoptions()

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