Io/tests/test_read_detector.py

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