diff options
Diffstat (limited to 'test/test_pdf.py')
| -rw-r--r-- | test/test_pdf.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_pdf.py b/test/test_pdf.py index 571cbd4..2eafd67 100644 --- a/test/test_pdf.py +++ b/test/test_pdf.py @@ -2,23 +2,23 @@ import unittest import numpy as np import itertools -import chroma.detectors +import chroma.demo from chroma.generator.photon import G4ParallelGenerator from chroma.generator.vertex import constant_particle_gun -from chroma.optics import water_wcsim +from chroma.demo.optics import water from chroma import gpu from chroma.sim import Simulation class TestPDF(unittest.TestCase): def setUp(self): - self.detector = chroma.detectors.microlbne() + self.detector = chroma.demo.tiny() self.detector.build() self.vertex_gen = constant_particle_gun('e-', (0,0,0), (1,0,0), 10) def testGPUPDF(self): '''Create a hit count and (q,t) PDF for 10 MeV events in MicroLBNE''' - g4generator = G4ParallelGenerator(1, water_wcsim) + g4generator = G4ParallelGenerator(1, water) context = gpu.create_cuda_context() |