tempestpy/tests/core/test_pla.py

import stormpy
import stormpy.logic
from helpers.helper import get_example_path

class TestModelChecking:
    def test_pla(self):
        import pycarl
        program = stormpy.parse_prism_program(get_example_path("pdtmc", "brp16_2.pm"))
        prop = "P<=0.84 [F s=5 ]"
        formulas = stormpy.parse_properties_for_prism_program(prop, program)
        model = stormpy.build_parametric_model(program, formulas)
        assert model.nr_states == 613
        assert model.nr_transitions == 803
        assert model.model_type == stormpy.ModelType.DTMC
        assert model.has_parameters
        checker = stormpy.PLAChecker(model)
        checker.specify_formula(formulas[0].raw_formula)
        parameters = model.collect_probability_parameters()
        assert len(parameters) == 2
        #if str(parameters[0]) == "pL":
        #    pL = parameters[0]
        #    pK = parameters[1]
        #else:
        #    pK = parameters[0]
        #    pL = parameters[1]
        #assert str(pL) == "pL"
        #assert str(pK) == "pK"
        #lowerBounds = {pL: 0.7, pK: 0.75}
        #upperBounds = {pL: 0.9, pK: 0.95}
        #region = stormpy.ParameterRegion(lowerBounds, upperBounds)
        region = stormpy.ParameterRegion("0.7<=pL<=0.9,0.75<=pK<=0.95", parameters)
        result = checker.check_region(region)
        assert result == stormpy.RegionCheckResult.ALLSAT
        #lowerBounds = {pL: 0.4, pK: 0.75}
        #upperBounds = {pL: 0.65, pK: 0.95}
        #region = stormpy.ParameterRegion(lowerBounds, upperBounds)
        region = stormpy.ParameterRegion("0.4<=pL<=0.65,0.75<=pK<=0.95", parameters)
        result = checker.check_region(region, stormpy.RegionCheckResult.UNKNOWN, True)
        assert result == stormpy.RegionCheckResult.EXISTSBOTH
        #lowerBounds = {pL: 0.1, pK: 0.2}
        #upperBounds = {pL: 0.73, pK: 0.715}
        #region = stormpy.ParameterRegion(lowerBounds, upperBounds)
        region = stormpy.ParameterRegion("0.1<=pL<=0.73,0.2<=pK<=0.715", parameters)
        result = checker.check_region(region)
        assert result == stormpy.RegionCheckResult.ALLVIOLATED