Core/tests/test_read_detector.py

0001 import os
0002 import tempfile
0003
0004 import detray.core
0005 import detray.examples
0006
0007
0008 def test_generate_and_read_detector():
0009     # Use the toy generator to produce detector files to read back.
0010     out_dir = tempfile.mkdtemp() + os.sep
0011     detray.examples.generateToyDetector(outputDir=out_dir)
0012
0013     geometry_file = os.path.join(out_dir, "toy_detector_geometry.json")
0014     assert os.path.exists(geometry_file)
0015
0016     detector, names = detray.core.readDetector(geometry_file)
0017     assert isinstance(names, detray.core.NameMap)
0018     assert detector.n_volumes() == 22
0019     assert detector.n_surfaces() == 3230