diff options
Diffstat (limited to 'test/test_rayleigh.py')
| -rw-r--r-- | test/test_rayleigh.py | 8 |
1 files changed, 4 insertions, 4 deletions
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) |