In [ ]:

    

import gamma_index
import numpy as np


    

In [ ]:

    

arr1 = np.random.rand(20, 20, 20)
arr2 = np.random.rand(20, 20, 20)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr2)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr2, 1, 0.05)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr2, 1, 0.03)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr2, 2, 0.03)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix(arr1, arr2, 2, 0.03)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr2, 3, 0.03)


    

In [ ]:

    

arr3 = arr1 + 0.03


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr1 + 0.03, 3, 0.03)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr1 + 0.1, 3, 0.03)


    

In [ ]:

    

%timeit gamma_index.gamma_matrix_c(arr1, arr1 + 0.01, 3, 0.03)


    

In [ ]: