Line data    Source code

       1             : #include "crpropa/ModuleList.h"
       2             : #include "crpropa/Source.h"
       3             : #include "crpropa/ParticleID.h"
       4             : #include "crpropa/module/SimplePropagation.h"
       5             : #include "crpropa/module/BreakCondition.h"
       6             : 
       7             : #include "gtest/gtest.h"
       8             : 
       9             : namespace crpropa {
      10             : 
      11           1 : TEST(ModuleList, process) {
      12           1 :         ModuleList modules;
      13           2 :         modules.add(new SimplePropagation());
      14           1 :         ParticleState initial;
      15           1 :         ref_ptr<Candidate> candidate = new Candidate(initial);
      16           2 :         modules.process(candidate);
      17           1 : }
      18             : 
      19           1 : TEST(ModuleList, getModule) {
      20           1 :         ModuleList modules;
      21           1 :         ref_ptr<SimplePropagation> prop = new SimplePropagation();
      22           1 :         modules.add(prop);
      23           2 :         EXPECT_TRUE(modules[0] == prop);
      24           1 : }
      25             : 
      26           1 : TEST(ModuleList, removeModule) {
      27           1 :         ModuleList modules;
      28           1 :         ref_ptr<SimplePropagation> prop = new SimplePropagation();
      29           1 :         modules.add(prop);
      30           1 :         modules.remove(0);
      31           1 :         EXPECT_EQ(modules.size(), 0);
      32           1 : }
      33             : 
      34           1 : TEST(ModuleList, runCandidateList) {
      35           1 :         ModuleList modules;
      36           2 :         modules.add(new SimplePropagation());
      37           1 :         modules.add(new MaximumTrajectoryLength(1 * Mpc));
      38           1 :         ParticleState initial;
      39           1 :         ref_ptr<Candidate> candidate = new Candidate(initial);
      40           1 :         modules.run(candidate);
      41           1 :         EXPECT_DOUBLE_EQ(1 * Mpc, candidate->getTrajectoryLength());
      42           1 :         EXPECT_TRUE(candidate->isActive() == false);
      43           1 : }
      44             : 
      45           1 : TEST(ModuleList, runSource) {
      46           1 :         ModuleList modules;
      47           2 :         modules.add(new SimplePropagation());
      48           1 :         modules.add(new MaximumTrajectoryLength(1 * Mpc));
      49             :         Source source;
      50           1 :         source.add(new SourcePosition(Vector3d(10, 0, 0) * Mpc));
      51           2 :         source.add(new SourceIsotropicEmission());
      52           1 :         source.add(new SourcePowerLawSpectrum(5 * EeV, 100 * EeV, -2));
      53           1 :         source.add(new SourceParticleType(nucleusId(1, 1)));
      54           1 :         modules.setShowProgress(true);
      55           1 :         modules.run(&source, 100, false);
      56           1 : }
      57             : 
      58             : #if _OPENMP
      59             : #include <omp.h>
      60           1 : TEST(ModuleList, runOpenMP) {
      61           1 :         ModuleList modules;
      62           2 :         modules.add(new SimplePropagation());
      63           1 :         modules.add(new MaximumTrajectoryLength(1 * Mpc));
      64             :         Source source;
      65           1 :         source.add(new SourcePosition(Vector3d(10, 0, 0) * Mpc));
      66           2 :         source.add(new SourceIsotropicEmission());
      67           1 :         source.add(new SourcePowerLawSpectrum(5 * EeV, 100 * EeV, -2));
      68           1 :         source.add(new SourceParticleType(nucleusId(1, 1)));
      69           1 :         omp_set_num_threads(2);
      70           1 :         modules.run(&source, 1000, false);
      71           1 : }
      72             : #endif
      73             : 
      74           0 : int main(int argc, char **argv) {
      75           0 :         ::testing::InitGoogleTest(&argc, argv);
      76           0 :         return RUN_ALL_TESTS();
      77             : }
      78             : 
      79             : } // namespace crpropa