New demo detector for Chroma that is not dependent on any internal experiment information.

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()