Method go_fast

tests/test_numba.py:13–17

(a)

        @jit(nopython=True) # Set "nopython" mode for best performance, equivalent to @njit
        def go_fast(a): # Function is compiled to machine code when called the first time
            trace = 0.0
            for i in range(a.shape[0]):   # Numba likes loops
                trace += np.tanh(a[i, i]) # Numba likes NumPy functions
            return a + trace              # Numba likes NumPy broadcasting

        self.assertEqual(10, go_fast(x).shape[0])