def test_digitization_example_input(trk_geo, tmp_path, assert_root_hash):
from particle_gun import runParticleGun
from digitization import configureDigitization
ptcl_dir = tmp_path / "ptcl"
ptcl_dir.mkdir()
pgs = Sequencer(events=20, numThreads=-1)
runParticleGun(str(ptcl_dir), s=pgs)
pgs.run()
s = Sequencer(numThreads=-1)
csv_dir = tmp_path / "csv"
root_file = tmp_path / "measurements.root"
assert not root_file.exists()
assert not csv_dir.exists()
assert_root_hash(
"particles.root",
ptcl_dir / "particles.root",
)
field = acts.ConstantBField(acts.Vector3(0, 0, 2 * u.T))
configureDigitization(
trk_geo,
field,
outputDir=tmp_path,
particlesInput=ptcl_dir / "particles.root",
s=s,
)
s.run()
assert root_file.exists()
assert csv_dir.exists()
assert len(list(csv_dir.iterdir())) == 3 * pgs.config.events
assert all(f.stat().st_size > 50 for f in csv_dir.iterdir())
for tn, nev in (
(7, 0),
(8, 193),
(9, 0),
(12, 1),
(13, 183),
(14, 6),
(16, 3),
(17, 76),
(18, 10),
):
assert_entries(root_file, f"vol{tn}", nev)
assert_root_hash(root_file.name, root_file)
def test_particle_gun(tmp_path, assert_root_hash):
from particle_gun import runParticleGun
s = Sequencer(events=20, numThreads=-1)
csv_dir = tmp_path / "csv"
root_file = tmp_path / "particles.root"
assert not csv_dir.exists()
assert not root_file.exists()
runParticleGun(str(tmp_path), s=s).run()
assert csv_dir.exists()
assert root_file.exists()
assert len([f for f in csv_dir.iterdir() if f.name.endswith("particles.csv")]) > 0
assert all([f.stat().st_size > 100 for f in csv_dir.iterdir()])
assert root_file.stat().st_size > 200
assert_entries(root_file, "particles", 20)
assert_root_hash(root_file.name, root_file)