qudarap/tests/QTexLookupTest.py

0001 #!/usr/bin/env python
0002 """
0003 ::
0004
0005     ipython -i tests/QTexLookupTest.py
0006
0007 """
0008 import os, logging, numpy as np
0009 log = logging.getLogger(__name__)
0010
0011
0012 class QTexLookupTest(object):
0013     FOLD = os.path.expandvars("/tmp/$USER/opticks/QTexLookupTest")
0014
0015     def __init__(self, fold=None):
0016         if fold is None:
0017             fold = self.FOLD
0018         pass
0019         names = os.listdir(fold)
0020         log.info("loading from fold %s " % fold)
0021         for name in filter(lambda _:_.endswith(".npy"), names):
0022             path = os.path.join(fold, name)
0023             stem = name[:-4]
0024             a = np.load(path)
0025             print( " t.%5s  %s " % (stem, str(a.shape)))
0026             setattr(self, stem, a )
0027             globals()[stem] = a
0028         pass
0029         self.fold = fold
0030     pass
0031
0032 if __name__ == '__main__':
0033      logging.basicConfig(level=logging.INFO)
0034      t = QTexLookupTest()
0035
0036      np.all( t.lookup == t.origin )
0037
0038