コード例 #1
0
ファイル: test.py プロジェクト: scienceopen/transcarread
def test_readtra():
#%% get sim parameters
    ifn   = tdir / 'data/DATCAR'
    tcofn = tdir / 'data/beam52.726/dir.output/transcar_output'
    tReq = datetime(2013,3,31,9,0,21,tzinfo=UTC)
    H = readTranscarInput(ifn)
#%% load transcar output
    iono,chi, pp = read_tra(tcofn,tReq)
#%% check
    assert_allclose(H['latgeo_ini'], 65.12)
    assert_allclose(iono.sel(param='n1')[30], 2.0969721e+11)
    assert_allclose(chi,110.40122986)
    assert_allclose(pp.sel(isrparam='Ti')[53], 1285.927001953125)
コード例 #2
0
def main(fn,tReq,verbose):
#%% get sim parameters
    datfn = join(split(split(fn)[0])[0],'dir.input/DATCAR')
    tctime = readTranscarInput(datfn)

    if isinstance(tReq,str):
        tReq = parse(tReq)

#%% load transcar output
    iono,chi, pp = read_tra(fn,tReq)
#%% do plot
    t = pp.time
    doPlot(t,iono,pp, fn, 'cubehelix',tctime,sfmt,verbose)

    #doplot1d(t,chi,'$\chi$',sfmt,fn, tctime)

    return iono,chi,pp,t,tctime