qudarap/tests/reflect_diffuse.py

0001 #!/usr/bin/env python
0002
0003 import os, numpy as np
0004 from opticks.ana.fold import Fold
0005
0006 def eprint( expr, lprefix="", rprefix="", tail="" ):
0007     ret = eval(expr)
0008     lhs = "%s%s" % (lprefix, expr)
0009     rhs = "%s%s" % (rprefix, ret )
0010     print("%-50s : %s%s" % ( lhs, rhs, tail )   )
0011     return ret
0012
0013 def epr(arg, **kwa):
0014     p = arg.find("=")
0015     if p > -1:
0016         var_eq = arg[:p+1]
0017         expr = arg[p+1:]
0018         label = var_eq
0019     else:
0020         label, expr = "", arg
0021     pass
0022     return eprint(expr, lprefix=label,  **kwa)
0023
0024
0025 if __name__ == '__main__':
0026     FOLD = os.environ["FOLD"]
0027     epr("FOLD")
0028     t = Fold.Load(FOLD)
0029
0030     if hasattr(t,'p0'):
0031         p0 = t.p0
0032         epr("p0", tail="\n\n", rprefix="\n"  )
0033     else:
0034         p0 = None
0035     pass
0036
0037     if hasattr(t,'prd'):
0038         prd = t.prd
0039         epr("prd", tail="\n\n", rprefix="\n" )
0040     else:
0041         prd = None
0042     pass
0043
0044     if hasattr(t,'s'):
0045         s = t.s
0046         epr("s", tail="\n\n", rprefix="\n" )
0047     else:
0048         s = None
0049     pass
0050
0051
0052     if hasattr(t,'p'):
0053         p = t.p
0054         epr("p", tail="\n\n", rprefix="\n" )
0055     else:
0056         p = None
0057     pass
0058
0059
0060     assert not p is None
0061
0062