sysrap/tests/SProfile_test.py

0001 #!/usr/bin/env python
0002
0003 import numpy as np
0004
0005 if __name__ == '__main__':
0006
0007     base = os.path.expandvars("$FOLD")
0008     name = "SProfile.npy"
0009     evts =  sorted(map(int,filter(lambda _:_.isdigit(), os.listdir(base))))
0010
0011     ee = {}
0012     for evt in evts:
0013         path = os.path.join(base, "%s"%evt, name)
0014         if not os.path.exists(path): continue
0015         ee[evt] = np.load(path)
0016     pass
0017
0018     for evt in evts:
0019         a = ee[evt]
0020         print("ee[%d] %s " % (evt, str(a.shape)) )
0021
0022         aa_ = "a[:,1:]-a[0,1]"
0023         aa = eval(aa_)
0024         print(aa_,repr(aa))
0025
0026         daa_ = "np.diff(aa.reshape(-1))"
0027         daa = eval(daa_)
0028         print(daa_,repr(daa))
0029     pass
0030
0031
0032 pass
0033