sysrap/tests/smath_test.py

0001 #!/usr/bin/env python
0002
0003 import os, textwrap, numpy as np
0004 from opticks.ana.fold import Fold, EXPR_
0005 from opticks.sysrap.smath import rotateUz
0006
0007 if __name__ == '__main__':
0008     f = Fold.Load(symbol="f")
0009     print(repr(f))
0010
0011     u = f.rotateUz[0,0]
0012     assert np.all( f.rotateUz[:,0] == u )
0013
0014     d = f.rotateUz[:,1]
0015     d1 = f.rotateUz[:,2]
0016     d1p = rotateUz(d, u)
0017
0018     for expr in EXPR_(r"""
0019 u
0020 d      # original direction from C++
0021 d1     # C++ rotateUz
0022 d1p    # py rotateUz
0023 d1 - d1p
0024 (d1 - d1p).min()
0025 (d1 - d1p).max()
0026 """):
0027         print(expr)
0028         if expr == "" or expr[0] == "#": continue
0029         print(repr(eval(expr)))
0030     pass
0031 pass
0032
0033