calc_para_grid.py 718 Bytes
import numpy as np
from time import time
from itertools import product

from odeg.h_eigenvalues import getEVs



if __name__ == '__main__':
    
    cfg = {
        'r0': 1,
        'rs': 2,
        'p_max': 5,
        'pos_charge': 10,
        }
    
    rs_grid = np.array([2., 4., 8.])
    r0_grid = np.array([.5, 1.])
    p_max_grid = np.array([3, 4, 5])
    
    
    for rs, r0, p_max in product(rs_grid, r0_grid, p_max_grid):
        cfg['rs'] = rs
        cfg['r0'] = r0
        cfg['p_max'] = p_max
        cfg['pos_charge'] = 2 * p_max
        
        
        print("\n\nconfiguration:\n", cfg, "\n\n")
        start = time()
        evs = getEVs(**cfg)
        print("t[fctexp] =", time() - start)