Python/tests/test_algorithms.py

0001 import pytest
0002
0003 from acts.examples import (
0004     AdaptiveMultiVertexFinderAlgorithm,
0005     VertexFitterAlgorithm,
0006     IterativeVertexFinderAlgorithm,
0007     HoughVertexFinderAlgorithm,
0008     SpacePointMaker,
0009     TrackFindingAlgorithm,
0010     SeedingAlgorithm,
0011     TrackParamsEstimationAlgorithm,
0012     EventGenerator,
0013     FatrasSimulation,
0014     MaterialMapping,
0015     TruthTrackFinder,
0016     ParticleSelector,
0017     TruthVertexFinder,
0018     TrackParameterSmearing,
0019     TrackSelectorAlgorithm,
0020     TrackFittingAlgorithm,
0021     ParticlesPrinter,
0022     TrackParametersPrinter,
0023     PropagationAlgorithm,
0024     DigitizationAlgorithm,
0025 )
0026
0027
0028 from helpers import geant4Enabled, hepmc3Enabled
0029
0030
0031 @pytest.mark.parametrize(
0032     "alg",
0033     [
0034         AdaptiveMultiVertexFinderAlgorithm,
0035         VertexFitterAlgorithm,
0036         IterativeVertexFinderAlgorithm,
0037         HoughVertexFinderAlgorithm,
0038         SpacePointMaker,
0039         TrackFindingAlgorithm,
0040         SeedingAlgorithm,
0041         TrackParamsEstimationAlgorithm,
0042         EventGenerator,
0043         FatrasSimulation,
0044         MaterialMapping,
0045         TruthTrackFinder,
0046         ParticleSelector,
0047         TruthVertexFinder,
0048         TrackParameterSmearing,
0049         TrackSelectorAlgorithm,
0050         TrackFittingAlgorithm,
0051         ParticlesPrinter,
0052         TrackParametersPrinter,
0053         PropagationAlgorithm,
0054         # GeantinoRecording,
0055         # EventRecording,
0056     ],
0057 )
0058 def test_algorithm_interface(alg):
0059     assert hasattr(alg, "Config")
0060
0061
0062 @pytest.mark.skipif(not geant4Enabled, reason="Geant4 not set up")
0063 def test_g4_algorithms():
0064     from acts.examples.geant4 import Geant4Simulation
0065
0066     assert hasattr(Geant4Simulation, "Config")
0067
0068
0069 def test_special_algorithm_interfaces():
0070     # just assert they exists
0071     assert DigitizationAlgorithm