Test momentum

Author: beomki-yeo

tests/integration_tests/cpu/propagator/propagator.cpp

@@ -508,6 +508,7 @@ TEST_P(PropagatorWithRkStepperDirectNavigator, direct_navigator) {
         // Propagate the entire detector
         // state.do_debug = true;
         ASSERT_TRUE(p.propagate(state, actor_states));
+        // std::cout << "Normal Navigation" << std::endl;
         // std::cout << state.debug_stream.str() << std::endl;
 
         if (seqs_device.size() > 0) {
@@ -518,8 +519,13 @@ TEST_P(PropagatorWithRkStepperDirectNavigator, direct_navigator) {
             direct_propagator_t::state direct_forward_state(track, bfield, det,
                                                             seqs_device);
 
+            // direct_forward_state.do_debug = true;
             ASSERT_TRUE(
                 direct_p.propagate(direct_forward_state, direct_actor_states));
+            // std::cout << "Direct navigaiton in forward direction" <<
+            // std::endl; std::cout << direct_forward_state.debug_stream.str()
+            // << std::endl;
+
             // Check if all surfaces in the sequence are encountered
             ASSERT_TRUE(direct_forward_state._navigation.is_complete());
 
@@ -529,6 +535,21 @@ TEST_P(PropagatorWithRkStepperDirectNavigator, direct_navigator) {
             direct_backward_state._navigation.set_direction(
                 detray::navigation::direction::e_backward);
 
+            const auto ptc = state._stepping.particle_hypothesis();
+            ASSERT_EQ(
+                ptc.pdg_num(),
+                direct_forward_state._stepping.particle_hypothesis().pdg_num());
+            const auto q = ptc.charge();
+
+            // The initial momentum should be higher than the momentum at the
+            // last surface
+            ASSERT_GE(track.p(q),
+                      static_cast<float>(state._stepping.bound_params().p(q)));
+            ASSERT_FLOAT_EQ(
+                static_cast<float>(state._stepping.bound_params().p(q)),
+                static_cast<float>(
+                    direct_forward_state._stepping.bound_params().p(q)));
+
             ASSERT_TRUE(
                 direct_p.propagate(direct_backward_state, direct_actor_states));
             // Check if all surfaces in the sequence are encountered