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)
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