diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..378eac2 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +build diff --git a/src/special_enrich_tests.h b/src/special_enrich_tests.h new file mode 100644 index 0000000..e37455f --- /dev/null +++ b/src/special_enrich_tests.h @@ -0,0 +1,52 @@ +#ifndef CYCVT_SRC_ENRICHMENT_TESTS_ +#define CYCVT_SRC_ENRICHMENT_TESTS_ + +#include + +#include + +#include "test_context.h" +#include "env.h" +#include "exchange_context.h" +#include "material.h" + +#include "special_enrich.h" + +namespace cycvt { + +// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - +class SEnrichmentTest : public ::testing::Test { + protected: + cyclus::TestContext tc_; + SEnrichment* src_facility; + std::string feed_commod, product_commod, feed_recipe, tails_commod; + cyclus::Composition::Ptr recipe; + TestFacility* trader; + + double feed_assay, tails_assay, inv_size, swu_capacity, max_enrich; + + bool order_prefs; + + double reserves; + + virtual void SetUp(); + virtual void TearDown(); + void InitParameters(); + void SetUpSource(); + cyclus::Material::Ptr GetMat(double qty); + /// @param enr the enrichment percent, i.e. for 5 w/o, enr = 0.05 + cyclus::Material::Ptr GetReqMat(double qty, double enr); + void DoAddMat(cyclus::Material::Ptr mat); + cyclus::Material::Ptr DoRequest(); + cyclus::Material::Ptr DoBid(cyclus::Material::Ptr mat); + cyclus::Material::Ptr DoOffer(cyclus::Material::Ptr mat); + cyclus::Material::Ptr DoEnrich(cyclus::Material::Ptr mat, double qty); + /// @param nreqs the total number of requests + /// @param nvalid the number of requests that are valid + boost::shared_ptr< cyclus::ExchangeContext > + GetContext(int nreqs, int nvalid); +}; + +} // namespace cycvt + +#endif // CYCVT_SRC_ENRICHMENT_FACILITY_TESTS_