1
0
Fork 0
mirror of https://github.com/gwm17/catima.git synced 2024-11-22 10:18:50 -05:00
catima/tests/test_structures.cpp
2020-08-05 18:20:16 +02:00

302 lines
10 KiB
C++

#define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
#define DOCTEST_CONFIG_SUPER_FAST_ASSERTS
#include "doctest.h"
#include "testutils.h"
#include <math.h>
using namespace std;
#include "catima/catima.h"
#include "catima/material_database.h"
TEST_CASE("atima material basic tests"){
catima::Material water({
{1,1,2},
{16,8,1}
});
catima::Material water2;
water2.add_element(1,1,2);
water2.add_element(16,8,1);
catima::Material graphite;
graphite.add_element(12,6,1);
graphite.density(1.8);
CHECK(graphite.ncomponents()==1);
CHECK(water.ncomponents()==2);
CHECK(graphite.M()==12);
CHECK(water.ncomponents()==2);
CHECK(water.M()==18);
CHECK(water2.ncomponents()==2);
CHECK(water2.M()==18);
CHECK(graphite.I()==0.0);
graphite.add_element(18,40,1);
CHECK(graphite.ncomponents()==2);
CHECK_FALSE(graphite.M()==approx(12,0.1));
CHECK( water.density() == approx(water2.density(),1e-6) );
CHECK(water==water2);
CHECK(!(water==graphite));
water.density(1.0);
water.thickness(1.0);
CHECK(water.thickness()==approx(1.0,0.0001));
water.thickness_cm(1.0);
CHECK(water.thickness()==approx(1.0,0.0001));
CHECK(water.thickness_cm()==approx(1.0,0.0001));
water.thickness_cm(2.0);
CHECK(water.thickness()==approx(2.0,0.0001));
CHECK(water.thickness_cm()==approx(2.0,0.0001));
water.density(2.0);
CHECK(water.thickness()==approx(2.0,0.0001));
CHECK(water.thickness_cm()==approx(1.0,0.0001));
water.thickness(1.0);
CHECK(water.thickness_cm()==approx(0.5,0.0001));
}
TEST_CASE("Material automatic atomic weight"){
catima::Material water({{0,1,2},{0,8,1}});
catima::Material graphite(0,6);
CHECK(water.get_element(0).A == 1.00794);
CHECK(water.get_element(1).A == 15.9994);
CHECK(graphite.get_element(0).A == 12.0107);
CHECK(water.M()>16);
CHECK(graphite.M()>12);
}
TEST_CASE("default materials"){
catima::Material m = catima::get_material(6);
CHECK(m.get_element(0).A == 12.0107);
CHECK(m.get_element(0).Z == 6);
CHECK(m.density() == 2.0);
CHECK(m.M() == 12.0107);
m = catima::get_material(catima::material::Water);
CHECK(m.get_element(0).A == 1.00794);
CHECK(m.get_element(0).Z == 1);
CHECK(m.get_element(1).A == 15.9994);
CHECK(m.get_element(1).Z == 8);
CHECK(m.density() == 1.0);
}
TEST_CASE("Layers"){
catima::Material water2;
water2.add_element(1,1,2);
water2.add_element(16,8,1);
water2.density(1.0).thickness(2.0);
catima::Material graphite;
graphite.add_element(12,6,1);
graphite.density(1.8).thickness(1.0);
catima::Layers detector1;
CHECK(detector1.num() == 0);
detector1.add(graphite);
CHECK(detector1.num() == 1);
detector1.add(water2);
detector1.add(graphite);
CHECK(detector1.num() == 3);
CHECK(detector1.thickness() == approx(4.0,1e-6));
// check correct density and thickness
CHECK(detector1[0].density()==1.8);
CHECK(detector1[0].thickness()==1.0);
CHECK(detector1[1].density()==1.0);
CHECK(detector1[1].thickness()==2.0);
CHECK(detector1[0].get_element(0).Z == 6);
CHECK(detector1[0].get_element(0).A == 12);
CHECK(detector1[1].get_element(0).A == 1);
CHECK(detector1[1].get_element(0).Z == 1);
CHECK(detector1[2].density() == 1.8);
detector1[1].density(1.2);
CHECK(detector1[1].density()==1.2);
catima::Layers detector2;
detector2 = detector1;
CHECK(detector2.num() == 3);
detector2.add(water2);
detector2.add(graphite);
CHECK(detector2.num() == 5);
catima::Layers focal_material = detector1 + detector2;
CHECK(focal_material.num() == 8);
}
TEST_CASE("basic projectile tests"){
catima::Projectile p{12,6,6,1000};
CHECK(p.A==12);
CHECK(p.Z==6);
CHECK(p.Q==6);
CHECK(p.T==1000);
catima::Projectile p2(12,6);
CHECK(p.A==12);
CHECK(p2.Z==6);
CHECK(p2.Q==6);
CHECK(p2.T==0);
p2(1000);
CHECK(p2.T==1000);
p2(1000)(500);
CHECK(p2.T==500);
catima::Projectile p3(12,6,5);
CHECK(p==p2);
CHECK( !(p==p3));
}
TEST_CASE("basic config test"){
catima::Config c1;
catima::Config c2;
catima::Config c3{catima::z_eff_type::none};
catima::Config c4;
CHECK(c1.z_effective == catima::default_config.z_effective);
CHECK(c1==c2);
CHECK( !(c1==c3));
CHECK(c1==c4);
c4.z_effective = catima::z_eff_type::global;
CHECK(!(c1==c4));
auto c5 = c4;
CHECK(c4==c5);
c4.z_effective = catima::z_eff_type::hubert;
CHECK(!(c4==c5) );
CHECK(!(c4==c1));
c4.z_effective = catima::default_config.z_effective;
CHECK(!(c5==c4));
CHECK((c1==c4));
}
TEST_CASE("constructors test"){
catima::Material mat2(12,6,2.5,0.1);
catima::Material mat3(12.01,6);
catima::Material mat4({{12.01,6,1.0}});
catima::Material mat5({
{12.01, 6, 1},
{16.00, 8, 2}
});
CHECK(mat2.ncomponents()==1);
CHECK(mat3.ncomponents()==1);
CHECK(mat3.get_element(0).A==12.01);
CHECK(mat4.ncomponents()==1);
CHECK(mat4.get_element(0).A==12.01);
CHECK(mat5.ncomponents()==2);
CHECK(mat5.get_element(0).A==12.01);
CHECK(mat5.get_element(0).Z==6);
CHECK(mat5.get_element(1).A==16.0);
CHECK(mat5.get_element(1).stn==2);
catima::Material mat6;
mat6 = mat5;
CHECK(mat5==mat6);
CHECK(mat5.density() == mat6.density());
CHECK(mat5.ncomponents()==mat6.ncomponents());
CHECK(mat5.get_element(0).A==mat6.get_element(0).A);
CHECK(mat5.get_element(1).A==mat6.get_element(1).A);
mat5.add_element(12,6,1);
CHECK(mat5.ncomponents()==mat6.ncomponents()+1);
// constructor with custom Ipot
catima::Material water1({
{1,1,2},
{16,8,1}
},1.0);
catima::Material water2({
{1,1,2},
{16,8,1}
},1.0, 78.0);
CHECK(water1.ncomponents()==2);
CHECK(water2.ncomponents()==2);
CHECK(water1.density()==1.0);
CHECK(water2.density()==1.0);
CHECK(water1.I()==0.0);
CHECK(water2.I()==78.0);
CHECK_FALSE(water1==water2);
}
TEST_CASE("fraction vs stn init"){
catima::Projectile p{12,6};
catima::Material water1({
{0, 1, 0.111894},
{0, 8, 0.888106}
});
catima::Material water2({
{0, 1, 2},
{0, 8, 1}
});
water1.thickness(1.0);
water2.thickness(1.0);
auto res1 = catima::calculate(p(600),water1);
auto res2 = catima::calculate(p(600),water2);
CHECK(res1.dEdxi == approx(res2.dEdxi,0.001));
CHECK(res1.range == approx(res2.range).R(1e-2));
CHECK(res1.sigma_a == approx(res2.sigma_a).R(1e-2));
CHECK(res1.sigma_r == approx(res2.sigma_r).R(1e-2));
}
TEST_CASE("fraction calculation"){
catima::Material water1({
{0, 1, 0.111898},
{0, 8, 0.888102}
});
catima::Material water2({
{0, 1, 2},
{0, 8, 1}
});
auto air = catima::get_material(catima::material::Air);
CHECK(water1.weight_fraction(0)==0.111898);
CHECK(water2.weight_fraction(0)==approx(water1.weight_fraction(0)).R(1e-5));
CHECK(water1.weight_fraction(1)==0.888102);
CHECK(water2.weight_fraction(1)==approx(water1.weight_fraction(1)).R(1e-5));
CHECK(water2.M()==approx(18).epsilon(0.1));
CHECK(water1.M()==approx(6.0,0.1));
CHECK(water2.M()==approx(18,0.1));
CHECK(water1.molar_fraction(0)==approx(2.0/3.0).R(1e-5));
CHECK(water2.molar_fraction(0)==approx(2.0).R(1e-5));
CHECK(water1.molar_fraction(1)==approx(1.0/3.0).R(1e-5));
CHECK(water2.molar_fraction(1)==approx(1.0).R(1e-5));
CHECK(water1.molar_fraction(0)/water1.molar_fraction(1)==approx(2.0).R(1e-5));
CHECK(water2.molar_fraction(0)/water2.molar_fraction(1)==approx(2.0).R(1e-5));
catima::Material mat({12.0,6,1});
CHECK(mat.M()==approx(12.0,0.001));
CHECK(mat.weight_fraction(0)==approx(1.0).R(1e-6));
//CHECK(air.M() == approx(28.97,0.1));
}
TEST_CASE("number density"){
catima::Material c({12.0,6,1});
auto water = catima::get_material(catima::material::Water);
auto air = catima::get_material(catima::material::Air);
water.density(0.9982);
c.density(3.513);
air.density(1.2041e-3);
c.thickness_cm(1.0);
CHECK(c.number_density()==approx(1.7662,0.01));
CHECK(c.number_density_cm2()==approx(1.7662,0.01));
CHECK(c.number_density(0)==approx(1.7662,0.01));
CHECK(c.number_density_cm2(0)==approx(1.7662,0.01));
CHECK(c.number_density(1)==0.0);
CHECK(c.number_density_cm2(1)==0.0);
c.thickness_cm(2.0);
CHECK(c.number_density()==approx(1.7662,0.01));
CHECK(c.number_density_cm2()==approx(2.0*1.7662,0.01));
water.thickness_cm(1.0);
CHECK(water.number_density()==approx(0.3336,0.001));
CHECK(water.number_density_cm2()==approx(0.3336,0.001));
CHECK(water.number_density(0)==approx(2*0.3336,0.001));
CHECK(water.number_density_cm2(0)==approx(2*0.3336,0.001));
CHECK(water.number_density(1)==approx(0.3336,0.001));
CHECK(water.number_density_cm2(1)==approx(0.3336,0.001));
water.thickness_cm(3.0);
CHECK(water.number_density_cm2()==approx(3.0*0.3336,0.001));
air.thickness_cm(1.0);
CHECK(air.number_density(0)==approx(air.molar_fraction(0)*2*0.0002504,0.00001));
CHECK(air.number_density(1)==approx(air.molar_fraction(1)*2*0.0002504,0.00001));
CHECK(air.number_density(2)==approx(air.molar_fraction(2)*1*0.0002504,0.00001));
}