diff --git a/src/test_suite.cpp b/src/test_suite.cpp index 41f8b9c..626aaba 100644 --- a/src/test_suite.cpp +++ b/src/test_suite.cpp @@ -9,7 +9,134 @@ * * @bug No known bugs * */ + +#include "PenningTrap.hpp" +#include "utils.hpp" + +#include +#include +#include + +void test_PenningTrap_external_E_field() +{ + PenningTrap trap; + + // Vector containing inputs and expected results + std::vector> tests; + + tests.push_back(std::make_pair(arma::vec{0.,0.,0.}, + arma::vec{0.,0.,0.})); + + tests.push_back(std::make_pair(arma::vec{10.,0.,0.}, + arma::vec{96.4852558,0.,0.})); + + tests.push_back(std::make_pair(arma::vec{10.,0.,0.}, + arma::vec{96.4852558,0.,0.})); + + tests.push_back(std::make_pair(arma::vec{0.,10.,0.}, + arma::vec{0.,96.4852558,0.})); + + tests.push_back(std::make_pair(arma::vec{0.,0.,10.}, + arma::vec{0.,0.,-192.9705116})); + + + arma::vec result; + arma::vec v; + std::stringstream msg; + for (int i=0; i < tests.size(); i++) { + v = tests.at(i).first; + result = trap.external_E_field(v); + + msg.str(""); + msg << "Testing the external E field at (" << std::setprecision(2) + << v(0) << "," << v(1) << "," << v(2) << ")."; + + ASSERT(arma_vector_close_to(result, tests.at(i).second), msg.str()); + } +} + + +void test_PenningTrap_external_B_field() +{ + // No point in testing at different points since it's not dependent on + // position. + PenningTrap trap; + arma::vec expected{0.,0.,T}; + arma::vec result = trap.external_B_field(arma::vec{0.,0.,0.}); + ASSERT(arma_vector_close_to(expected, result), + "Testing the external B field at (0,0,0)"); +} + +void test_PenningTrap_force_on_particle() +{ + PenningTrap trap; + arma::vec v{0.,0.,0.}; + + // Add particles to test + trap.add_particle(Particle(1.,40.,arma::vec{0.,0.,0.},v)); + trap.add_particle(Particle(1.,40.,arma::vec{1.,0.,0.},v)); + trap.add_particle(Particle(1.,40.,arma::vec{0.,3.,4.},v)); + + // Test p0 and p1 + arma::vec expected{-1.,0.,0.}; + arma::vec result = trap.force_on_particle(0, 1); + ASSERT(arma_vector_close_to(expected, result), + "Testing the force on a particle at (0,0,0) from a " + "particle at (1,0,0)."); + + // Test p0 and p2 + expected = arma::vec{0, -.024, -.032}; + result = trap.force_on_particle(0, 2); + ASSERT(arma_vector_close_to(expected, result), + "Testing the force on a particle at (0,0,0) from a " + "particle at (0,3,4)."); +} + +void test_PenningTrap_total_force_external() +{ + PenningTrap trap; + trap.add_particle(Particle(1.,40.,arma::vec{1.,2.,3.}, + arma::vec{3.,4.,5.})); + + arma::vec expected{395.58954878,-270.15871624,-57.89115348}; + arma::vec result = trap.total_force_external(0); + ASSERT(arma_vector_close_to(expected, result), + "Testing the total external force on a particle at " + "(1,2,3) with velocity (3,4,5)"); +} + +void test_PenningTrap_total_force_particles() +{ + PenningTrap trap; + trap.add_particle(Particle(1.,40.,arma::vec{0.,0.,0.}, + arma::vec{0.,0.,0.})); + + arma::vec expected{0.,0.,0.}; + arma::vec result = trap.total_force_particles(0); + ASSERT(arma_vector_close_to(expected, result), + "Testing the total force of all particles on particle 0 " + "with only a single particle"); + + trap.add_particle(Particle(1.,40.,arma::vec{1.,0.,0.}, + arma::vec{0.,0.,0.})); + trap.add_particle(Particle(1.,40.,arma::vec{0.,1.,0.}, + arma::vec{0.,0.,0.})); + trap.add_particle(Particle(1.,40.,arma::vec{0.,0.,1.}, + arma::vec{0.,0.,0.})); + + expected = arma::vec(3,arma::fill::value(-3473.383325)); + result = trap.total_force_particles(0); + ASSERT(arma_vector_close_to(expected, result), + "Testing the total force of all particles on particle 0 " + "with 3 other particles."); +} + int main() { + test_PenningTrap_external_E_field(); + test_PenningTrap_external_B_field(); + test_PenningTrap_force_on_particle(); + test_PenningTrap_total_force_external(); + test_PenningTrap_total_force_particles(); return 0; }