diff options
| -rw-r--r-- | chroma/.hgsub | 1 | ||||
| -rw-r--r-- | test/test_rayleigh.py | 8 |
2 files changed, 5 insertions, 4 deletions
diff --git a/chroma/.hgsub b/chroma/.hgsub new file mode 100644 index 0000000..d3f70fc --- /dev/null +++ b/chroma/.hgsub @@ -0,0 +1 @@ +histogram = http://bitbucket.org/tlatorre/histogram diff --git a/test/test_rayleigh.py b/test/test_rayleigh.py index cfb0441..487184f 100644 --- a/test/test_rayleigh.py +++ b/test/test_rayleigh.py @@ -6,11 +6,11 @@ from chroma.make import box from chroma.sim import Simulation from chroma.demo.optics import water from chroma.event import Photons -import histogram -from histogram.root import rootify -import ROOT +from chroma.rootimport import ROOT ROOT.gROOT.SetBatch(1) +from chroma.histogram import Histogram, rootify + class TestRayleigh(unittest.TestCase): def setUp(self): self.cube = Geometry(water) @@ -44,7 +44,7 @@ class TestRayleigh(unittest.TestCase): rayleigh_scatters = (photons_end.flags & (1 << 4)) > 0 cos_scatter = (self.photons.dir[rayleigh_scatters] * photons_end.dir[rayleigh_scatters]).sum(axis=1) theta_scatter = np.arccos(cos_scatter) - h = histogram.Histogram(bins=100, range=(0, np.pi)) + h = Histogram(bins=100, range=(0, np.pi)) h.fill(theta_scatter) h = rootify(h) |