From 54433ca8a38f155003e1040d97c25dcf24ee5939 Mon Sep 17 00:00:00 2001
From: Matthias Volk <matthias.volk@cs.rwth-aachen.de>
Date: Fri, 18 May 2018 11:53:50 +0200
Subject: [PATCH] Tests for symbolic model building

---
 tests/storage/test_model.py | 87 +++++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)

diff --git a/tests/storage/test_model.py b/tests/storage/test_model.py
index 12b8326..2598e3b 100644
--- a/tests/storage/test_model.py
+++ b/tests/storage/test_model.py
@@ -123,3 +123,90 @@ class TestSparseModel:
         initial_states = model.initial_states
         assert len(initial_states) == 1
         assert 0 in initial_states
+
+
+class TestSymbolicSylvanModel:
+    def test_build_dtmc_from_prism_program(self):
+        program = stormpy.parse_prism_program(get_example_path("dtmc", "die.pm"))
+        model = stormpy.build_symbolic_model(program)
+        assert model.nr_states == 13
+        assert model.nr_transitions == 20
+        assert model.model_type == stormpy.ModelType.DTMC
+        assert not model.supports_parameters
+        assert type(model) is stormpy.SymbolicSylvanDtmc
+
+    def test_build_dtmc_from_prism_program_formulas(self):
+        program = stormpy.parse_prism_program(get_example_path("dtmc", "die.pm"))
+        prop = "P=? [F \"one\"]"
+        properties = stormpy.parse_properties_for_prism_program(prop, program, None)
+        model = stormpy.build_symbolic_model(program, properties)
+        assert model.nr_states == 13
+        assert model.nr_transitions == 20
+        assert model.model_type == stormpy.ModelType.DTMC
+        assert len(model.reward_models) == 0
+        assert not model.supports_parameters
+        assert type(model) is stormpy.SymbolicSylvanDtmc
+
+    def test_build_dtmc_from_prism_program_reward_formulas(self):
+        program = stormpy.parse_prism_program(get_example_path("dtmc", "die.pm"))
+        prop = "R=? [F \"done\"]"
+        properties = stormpy.parse_properties_for_prism_program(prop, program, None)
+        model = stormpy.build_symbolic_model(program, properties)
+        assert model.nr_states == 13
+        assert model.nr_transitions == 20
+        assert model.model_type == stormpy.ModelType.DTMC
+        assert len(model.reward_models) == 1
+        assert not model.reward_models["coin_flips"].has_state_rewards
+        assert model.reward_models["coin_flips"].has_state_action_rewards
+        assert not model.reward_models["coin_flips"].has_transition_rewards
+        assert not model.supports_parameters
+        assert type(model) is stormpy.SymbolicSylvanDtmc
+
+    def test_reduce_to_state_based_rewards(self):
+        program = stormpy.parse_prism_program(get_example_path("dtmc", "die.pm"))
+        prop = "R=? [F \"done\"]"
+        properties = stormpy.parse_properties_for_prism_program(prop, program, None)
+        model = stormpy.build_symbolic_model(program, properties)
+        model.reduce_to_state_based_rewards()
+        assert len(model.reward_models) == 1
+        assert model.reward_models["coin_flips"].has_state_rewards
+        assert not model.reward_models["coin_flips"].has_state_action_rewards
+        assert not model.reward_models["coin_flips"].has_transition_rewards
+
+    def test_build_dtmc_from_jani_model(self):
+        jani_model, properties = stormpy.parse_jani_model(get_example_path("dtmc", "brp.jani"))
+        description = stormpy.SymbolicModelDescription(jani_model)
+        constant_definitions = description.parse_constant_definitions("N=16, MAX=2")
+        instantiated_jani_model = description.instantiate_constants(constant_definitions).as_jani_model()
+        model = stormpy.build_symbolic_model(instantiated_jani_model)
+        assert model.nr_states == 677
+        assert model.nr_transitions == 867
+        assert model.model_type == stormpy.ModelType.DTMC
+        assert not model.supports_parameters
+        assert type(model) is stormpy.SymbolicSylvanDtmc
+
+    def test_build_mdp(self):
+        program = stormpy.parse_prism_program(get_example_path("mdp", "two_dice.nm"))
+        formulas = stormpy.parse_properties_for_prism_program("P=? [ F \"two\" ]", program)
+        model = stormpy.build_symbolic_model(program, formulas)
+        assert model.nr_states == 169
+        assert model.nr_transitions == 435
+        assert model.model_type == stormpy.ModelType.MDP
+        assert not model.supports_parameters
+        assert type(model) is stormpy.SymbolicSylvanMdp
+
+    def test_build_ctmc(self):
+        program = stormpy.parse_prism_program(get_example_path("ctmc", "polling2.sm"), True)
+        formulas = stormpy.parse_properties_for_prism_program("P=? [ F<=3 \"target\" ]", program)
+        model = stormpy.build_symbolic_model(program, formulas)
+        assert model.nr_states == 12
+        assert model.nr_transitions == 22
+        assert model.model_type == stormpy.ModelType.CTMC
+        assert not model.supports_parameters
+        assert type(model) is stormpy.SymbolicSylvanCtmc
+
+    def test_build_ma(self):
+        program = stormpy.parse_prism_program(get_example_path("ma", "simple.ma"))
+        formulas = stormpy.parse_properties_for_prism_program("P=? [ F<=2 s=2 ]", program)
+        with pytest.raises(Exception):
+            model = stormpy.build_symbolic_model(program, formulas)