You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
3.7 KiB
72 lines
3.7 KiB
#include "gtest/gtest.h"
|
|
#include "storm-config.h"
|
|
|
|
#include "storm/parser/DirectEncodingParser.h"
|
|
#include "storm/models/sparse/StandardRewardModel.h"
|
|
#include "storm/models/sparse/Mdp.h"
|
|
#include "storm/models/sparse/MarkovAutomaton.h"
|
|
|
|
TEST(DirectEncodingParserTest, DtmcParsing) {
|
|
std::shared_ptr<storm::models::sparse::Model<double>> modelPtr = storm::parser::DirectEncodingParser<double>::parseModel(STORM_TEST_RESOURCES_DIR "/dtmc/crowds-5-5.drn");
|
|
|
|
// Test if parsed correctly.
|
|
ASSERT_EQ(storm::models::ModelType::Dtmc, modelPtr->getType());
|
|
ASSERT_EQ(8607ul, modelPtr->getNumberOfStates());
|
|
ASSERT_EQ(15113ul, modelPtr->getNumberOfTransitions());
|
|
ASSERT_TRUE(modelPtr->hasLabel("init"));
|
|
ASSERT_EQ(1ul, modelPtr->getInitialStates().getNumberOfSetBits());
|
|
ASSERT_TRUE(modelPtr->hasLabel("observeIGreater1"));
|
|
ASSERT_EQ(4650ul, modelPtr->getStates("observeIGreater1").getNumberOfSetBits());
|
|
ASSERT_TRUE(modelPtr->hasLabel("observe0Greater1"));
|
|
ASSERT_EQ(1260ul, modelPtr->getStates("observe0Greater1").getNumberOfSetBits());
|
|
}
|
|
|
|
TEST(DirectEncodingParserTest, MdpParsing) {
|
|
std::shared_ptr<storm::models::sparse::Model<double>> modelPtr = storm::parser::DirectEncodingParser<double>::parseModel(STORM_TEST_RESOURCES_DIR "/mdp/two_dice.drn");
|
|
|
|
// Test if parsed correctly.
|
|
ASSERT_EQ(storm::models::ModelType::Mdp, modelPtr->getType());
|
|
ASSERT_EQ(169ul, modelPtr->getNumberOfStates());
|
|
ASSERT_EQ(436ul, modelPtr->getNumberOfTransitions());
|
|
ASSERT_EQ(254ul, modelPtr->as<storm::models::sparse::Mdp<double>>()->getNumberOfChoices());
|
|
ASSERT_TRUE(modelPtr->hasLabel("init"));
|
|
ASSERT_EQ(1ul, modelPtr->getInitialStates().getNumberOfSetBits());
|
|
ASSERT_TRUE(modelPtr->hasLabel("six"));
|
|
ASSERT_EQ(5ul, modelPtr->getStates("six").getNumberOfSetBits());
|
|
ASSERT_TRUE(modelPtr->hasLabel("eleven"));
|
|
ASSERT_EQ(2ul, modelPtr->getStates("eleven").getNumberOfSetBits());
|
|
}
|
|
|
|
TEST(DirectEncodingParserTest, CtmcParsing) {
|
|
std::shared_ptr<storm::models::sparse::Model<double>> modelPtr = storm::parser::DirectEncodingParser<double>::parseModel(STORM_TEST_RESOURCES_DIR "/ctmc/cluster2.drn");
|
|
|
|
// Test if parsed correctly.
|
|
ASSERT_EQ(storm::models::ModelType::Ctmc, modelPtr->getType());
|
|
ASSERT_EQ(276ul, modelPtr->getNumberOfStates());
|
|
ASSERT_EQ(1120ul, modelPtr->getNumberOfTransitions());
|
|
ASSERT_TRUE(modelPtr->hasLabel("init"));
|
|
ASSERT_EQ(1ul, modelPtr->getInitialStates().getNumberOfSetBits());
|
|
ASSERT_TRUE(modelPtr->hasLabel("premium"));
|
|
ASSERT_EQ(64ul, modelPtr->getStates("premium").getNumberOfSetBits());
|
|
ASSERT_TRUE(modelPtr->hasLabel("minimum"));
|
|
ASSERT_EQ(132ul, modelPtr->getStates("minimum").getNumberOfSetBits());
|
|
}
|
|
|
|
TEST(DirectEncodingParserTest, MarkovAutomatonParsing) {
|
|
std::shared_ptr<storm::models::sparse::Model<double>> modelPtr = storm::parser::DirectEncodingParser<double>::parseModel(STORM_TEST_RESOURCES_DIR "/ma/jobscheduler.drn");
|
|
std::shared_ptr<storm::models::sparse::MarkovAutomaton<double>> ma = modelPtr->as<storm::models::sparse::MarkovAutomaton<double>>();
|
|
|
|
// Test if parsed correctly.
|
|
ASSERT_EQ(storm::models::ModelType::MarkovAutomaton, modelPtr->getType());
|
|
ASSERT_EQ(17ul, ma->getNumberOfStates());
|
|
ASSERT_EQ(25ul, ma->getNumberOfTransitions());
|
|
ASSERT_EQ(19ul, ma->getNumberOfChoices());
|
|
ASSERT_EQ(10ul, ma->getMarkovianStates().getNumberOfSetBits());
|
|
ASSERT_EQ(5, ma->getMaximalExitRate());
|
|
ASSERT_TRUE(ma->hasRewardModel("avg_waiting_time"));
|
|
ASSERT_TRUE(modelPtr->hasLabel("init"));
|
|
ASSERT_EQ(1ul, modelPtr->getInitialStates().getNumberOfSetBits());
|
|
ASSERT_TRUE(modelPtr->hasLabel("one_job_finished"));
|
|
ASSERT_EQ(6ul, modelPtr->getStates("one_job_finished").getNumberOfSetBits());
|
|
}
|
|
|