Browse Source

Added first working test case

tempestpy_adaptions
Tim Quatmann 5 years ago
parent
commit
202c25c3db
  1. 48
      src/test/storm-pomdp/modelchecker/ApproximatePOMDPModelcheckerTest.cpp

48
src/test/storm-pomdp/modelchecker/ApproximatePOMDPModelcheckerTest.cpp

@ -2,6 +2,7 @@
#include "storm-config.h" #include "storm-config.h"
#include "storm-pomdp/modelchecker/ApproximatePOMDPModelchecker.h" #include "storm-pomdp/modelchecker/ApproximatePOMDPModelchecker.h"
#include "storm-pomdp/transformer/MakePOMDPCanonic.h"
#include "storm/api/storm.h" #include "storm/api/storm.h"
#include "storm-parsers/api/storm-parsers.h" #include "storm-parsers/api/storm-parsers.h"
@ -126,15 +127,35 @@ namespace {
TestType::adaptOptions(opt); TestType::adaptOptions(opt);
return opt; return opt;
} }
void preprocess(std::shared_ptr<storm::models::sparse::Pomdp<ValueType>>& model, std::shared_ptr<const storm::logic::Formula>& formula) {
switch(TestType::preprocessingType) {
ValueType parseNumber(std::string const& str) {
return storm::utility::convertNumber<ValueType>(str);
}
struct Input {
std::shared_ptr<storm::models::sparse::Pomdp<ValueType>> model;
std::shared_ptr<storm::logic::Formula const> formula;
};
Input buildPrism(std::string const& programFile, std::string const& formulaAsString, std::string const& constantsAsString = "") const {
// Parse and build input
storm::prism::Program program = storm::api::parseProgram(programFile);
program = storm::utility::prism::preprocess(program, constantsAsString);
Input input;
input.formula = storm::api::parsePropertiesForPrismProgram(formulaAsString, program).front().getRawFormula();
input.model = storm::api::buildSparseModel<ValueType>(program, {input.formula})->template as<storm::models::sparse::Pomdp<ValueType>>();
// Preprocess
storm::transformer::MakePOMDPCanonic<ValueType> makeCanonic(*input.model);
input.model = makeCanonic.transform();
EXPECT_TRUE(input.model->isCanonic());
switch (TestType::preprocessingType) {
case PreprocessingType::None: case PreprocessingType::None:
// nothing to do // nothing to do
break; break;
default: default:
FAIL() << "Unhandled preprocessing type.";
ADD_FAILURE() << "Unhandled preprocessing type.";
} }
EXPECT_TRUE(input.model->isCanonic());
return input;
} }
private: private:
storm::Environment _environment; storm::Environment _environment;
}; };
@ -152,24 +173,15 @@ namespace {
TYPED_TEST_SUITE(BeliefExplorationTest, TestingTypes,); TYPED_TEST_SUITE(BeliefExplorationTest, TestingTypes,);
TYPED_TEST(BeliefExplorationTest, simple) {
TYPED_TEST(BeliefExplorationTest, simple_max) {
typedef typename TestFixture::ValueType ValueType; typedef typename TestFixture::ValueType ValueType;
std::string programFile = STORM_TEST_RESOURCES_DIR "/pomdp/simple.prism";
std::string formulaAsString = "P=? [F \"goal\" ]";
std::string constantsAsString = ""; //e.g. pL=0.9,TOACK=0.5
auto data = this->buildPrism(STORM_TEST_RESOURCES_DIR "/pomdp/simple.prism", "Pmax=? [F \"goal\" ]", "slippery=0");
storm::pomdp::modelchecker::ApproximatePOMDPModelchecker<storm::models::sparse::Pomdp<ValueType>> checker(data.model, this->options());
auto result = checker.check(*data.formula);
// Program and formula
storm::prism::Program program = storm::api::parseProgram(programFile);
program = storm::utility::prism::preprocess(program, constantsAsString);
std::shared_ptr<const storm::logic::Formula> formula = storm::api::parsePropertiesForPrismProgram(formulaAsString, program).front().getRawFormula();
std::shared_ptr<storm::models::sparse::Pomdp<ValueType>> model = storm::api::buildSparseModel<ValueType>(program, {formula})->template as<storm::models::sparse::Pomdp<ValueType>>();
this->preprocess(model, formula);
// Invoke model checking
storm::pomdp::modelchecker::ApproximatePOMDPModelchecker<storm::models::sparse::Pomdp<ValueType>> checker(*model, this->options());
auto result = checker.check(*formula);
std::cout << "[" << result.lowerBound << "," << result.upperBound << std::endl;
EXPECT_LE(result.lowerBound, this->parseNumber("7/10"));
EXPECT_GE(result.upperBound, this->parseNumber("7/10"));
} }
} }
Loading…
Cancel
Save