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.
252 lines
9.1 KiB
252 lines
9.1 KiB
import os
|
|
|
|
import stormpy
|
|
|
|
from configurations import gspn
|
|
|
|
|
|
@gspn
|
|
class TestGSPNBuilder:
|
|
def test_layout_info(self):
|
|
layout = stormpy.gspn.LayoutInfo()
|
|
assert layout.x == 0
|
|
assert layout.y == 0
|
|
assert layout.rotation == 0
|
|
layout.x = 1
|
|
assert layout.x == 1
|
|
|
|
layout_xy = stormpy.gspn.LayoutInfo(2, 3)
|
|
assert layout_xy.x == 2
|
|
assert layout_xy.rotation == 0
|
|
|
|
layout_xyr = stormpy.gspn.LayoutInfo(2, 3, 4)
|
|
assert layout_xyr.rotation == 4
|
|
|
|
def test_place(self):
|
|
p_id = 4
|
|
place = stormpy.gspn.Place(id=p_id)
|
|
assert p_id == place.get_id()
|
|
|
|
assert not place.has_restricted_capacity()
|
|
place.set_capacity(cap=5)
|
|
assert place.has_restricted_capacity()
|
|
assert place.get_capacity() == 5
|
|
|
|
p_name = "P_0"
|
|
place.set_name(name=p_name)
|
|
assert place.get_name() == p_name
|
|
|
|
p_tokens = 2
|
|
place.set_number_of_initial_tokens(p_tokens)
|
|
assert place.get_number_of_initial_tokens() == p_tokens
|
|
|
|
def test_transition(self):
|
|
# test TimedTransition
|
|
tt = stormpy.gspn.TimedTransition()
|
|
tt_name = " tt"
|
|
tt.set_name(tt_name)
|
|
assert tt_name == tt.get_name()
|
|
tt_rate = 0.2
|
|
tt.set_rate(tt_rate)
|
|
assert tt_rate == tt.get_rate()
|
|
|
|
# connect a place to this transition and test arcs
|
|
place = stormpy.gspn.Place(0)
|
|
# test input arcs
|
|
assert not tt.exists_input_arc(place)
|
|
tt.set_input_arc_multiplicity(place, 2)
|
|
assert tt.exists_input_arc(place)
|
|
assert tt.get_input_arc_multiplicity(place) == 2
|
|
tt.remove_input_arc(place)
|
|
assert not tt.exists_input_arc(place)
|
|
# test output arcs
|
|
assert not tt.exists_output_arc(place)
|
|
tt.set_output_arc_multiplicity(place, 3)
|
|
assert tt.exists_output_arc(place)
|
|
assert tt.get_output_arc_multiplicity(place) == 3
|
|
tt.remove_output_arc(place)
|
|
assert not tt.exists_output_arc(place)
|
|
# test inhibition arcs
|
|
assert not tt.exists_inhibition_arc(place)
|
|
tt.set_inhibition_arc_multiplicity(place, 5)
|
|
assert tt.exists_inhibition_arc(place)
|
|
assert tt.get_inhibition_arc_multiplicity(place) == 5
|
|
tt.remove_inhibition_arc(place)
|
|
assert not tt.exists_inhibition_arc(place)
|
|
|
|
# test ImmediateTransition
|
|
ti = stormpy.gspn.ImmediateTransition()
|
|
ti_name = " ti"
|
|
ti.set_name(ti_name)
|
|
assert ti_name == ti.get_name()
|
|
assert ti.no_weight_attached()
|
|
ti_weight = 0.2
|
|
ti.set_weight(ti_weight)
|
|
assert ti_weight == ti.get_weight()
|
|
assert not ti.no_weight_attached()
|
|
|
|
def test_build_gspn(self):
|
|
gspn_name = "gspn_test"
|
|
builder = stormpy.gspn.GSPNBuilder()
|
|
|
|
id_p_0 = builder.add_place()
|
|
id_p_1 = builder.add_place(initial_tokens=1)
|
|
id_p_2 = builder.add_place(initial_tokens=0, name="place_2")
|
|
id_p_3 = builder.add_place(capacity=2, initial_tokens=3, name="place_3")
|
|
p_layout = stormpy.gspn.LayoutInfo(1, 2)
|
|
builder.set_place_layout_info(id_p_0, p_layout)
|
|
|
|
id_ti_0 = builder.add_immediate_transition(priority=1, weight=0.5, name="ti_0")
|
|
id_ti_1 = builder.add_immediate_transition()
|
|
|
|
id_tt_0 = builder.add_timed_transition(priority=2, rate=0.4, name="tt_0")
|
|
id_tt_1 = builder.add_timed_transition(priority=0, rate=0.5, num_servers=2, name="tt_1")
|
|
|
|
t_layout = stormpy.gspn.LayoutInfo(1, 2)
|
|
builder.set_transition_layout_info(id_ti_0, t_layout)
|
|
|
|
# add arcs
|
|
builder.add_input_arc(id_p_0, id_ti_1, multiplicity=2)
|
|
builder.add_input_arc("place_2", "ti_0", multiplicity=2)
|
|
|
|
builder.add_output_arc(id_ti_1, id_p_2, multiplicity=2)
|
|
builder.add_output_arc("tt_0", "place_3", multiplicity=2)
|
|
|
|
builder.add_inhibition_arc(id_p_2, id_tt_0, multiplicity=2)
|
|
builder.add_inhibition_arc("place_3", "tt_0", multiplicity=2)
|
|
|
|
builder.add_normal_arc("place_3", "tt_0", multiplicity=2)
|
|
builder.add_normal_arc("tt_0", "place_3", multiplicity=2)
|
|
|
|
# test gspn composition
|
|
builder.set_name(gspn_name)
|
|
gspn = builder.build_gspn()
|
|
|
|
assert gspn.get_name() == gspn_name
|
|
assert gspn.is_valid()
|
|
assert gspn.get_number_of_immediate_transitions() == 2
|
|
assert gspn.get_number_of_timed_transitions() == 2
|
|
assert gspn.get_number_of_places() == 4
|
|
|
|
assert len(gspn.get_places()) == 4
|
|
assert len(gspn.get_immediate_transitions()) == 2
|
|
assert len(gspn.get_timed_transitions()) == 2
|
|
|
|
# test places
|
|
p_0 = gspn.get_place(id_p_0)
|
|
assert p_0.get_id() == id_p_0
|
|
|
|
p_1 = gspn.get_place(id_p_1)
|
|
assert p_1.get_id() == id_p_1
|
|
assert p_1.get_number_of_initial_tokens() == 1
|
|
|
|
p_2 = gspn.get_place(id_p_2)
|
|
assert p_2.get_id() == id_p_2
|
|
assert p_2.get_name() == "place_2"
|
|
|
|
p_3 = gspn.get_place(id_p_3)
|
|
assert p_3.get_name() == "place_3"
|
|
assert p_3.get_capacity() == 2
|
|
assert p_3.get_number_of_initial_tokens() == 3
|
|
assert p_3.has_restricted_capacity()
|
|
|
|
# test transitions
|
|
ti_0 = gspn.get_immediate_transition("ti_0")
|
|
assert ti_0.get_id() == id_ti_0
|
|
assert ti_0.get_weight() == 0.5
|
|
assert ti_0.get_priority() == 1
|
|
|
|
tt_0 = gspn.get_timed_transition("tt_0")
|
|
assert tt_0.get_id() == id_tt_0
|
|
assert tt_0.get_rate() == 0.4
|
|
assert tt_0.get_priority() == 2
|
|
|
|
tt_1 = gspn.get_timed_transition("tt_1")
|
|
assert tt_1.get_id() == id_tt_1
|
|
assert tt_1.get_number_of_servers() == 2
|
|
|
|
# test new name
|
|
gspn_new_name = "new_name"
|
|
gspn.set_name(gspn_new_name)
|
|
assert gspn.get_name() == gspn_new_name
|
|
|
|
def test_export_to_pnpro(self, tmpdir):
|
|
builder = stormpy.gspn.GSPNBuilder()
|
|
builder.set_name("gspn_test")
|
|
|
|
# add places and transitions
|
|
id_p_0 = builder.add_place()
|
|
id_p_1 = builder.add_place(initial_tokens=3, name="place_1", capacity=2)
|
|
id_ti_0 = builder.add_immediate_transition(priority=0, weight=0.5, name="ti_0")
|
|
id_tt_0 = builder.add_timed_transition(priority=0, rate=0.5, num_servers=2, name="tt_0")
|
|
|
|
gspn = builder.build_gspn()
|
|
|
|
export_file = os.path.join(str(tmpdir), "gspn.pnpro")
|
|
|
|
# export gspn to pnml
|
|
gspn.export_gspn_pnpro_file(export_file)
|
|
|
|
# import gspn
|
|
gspn_parser = stormpy.gspn.GSPNParser()
|
|
gspn_import = gspn_parser.parse(export_file)
|
|
|
|
# test imported gspn
|
|
assert gspn_import.get_name() == gspn.get_name()
|
|
assert gspn_import.get_number_of_timed_transitions() == gspn.get_number_of_timed_transitions()
|
|
assert gspn_import.get_number_of_immediate_transitions() == gspn.get_number_of_immediate_transitions()
|
|
assert gspn_import.get_number_of_places() == gspn.get_number_of_places()
|
|
|
|
p_0 = gspn_import.get_place(id_p_0)
|
|
assert p_0.get_id() == id_p_0
|
|
p_1 = gspn_import.get_place(id_p_1)
|
|
assert p_1.get_name() == "place_1"
|
|
# todo capacity info lost
|
|
# assert p_1.get_capacity() == 2
|
|
# assert p_1.has_restricted_capacity() == True
|
|
assert p_1.get_number_of_initial_tokens() == 3
|
|
|
|
ti_0 = gspn_import.get_immediate_transition("ti_0")
|
|
assert ti_0.get_id() == id_ti_0
|
|
tt_0 = gspn_import.get_timed_transition("tt_0")
|
|
assert tt_0.get_id() == id_tt_0
|
|
|
|
def test_export_to_pnml(self, tmpdir):
|
|
builder = stormpy.gspn.GSPNBuilder()
|
|
builder.set_name("gspn_test")
|
|
|
|
# add places and transitions
|
|
id_p_0 = builder.add_place()
|
|
id_p_1 = builder.add_place(initial_tokens=3, name="place_1", capacity=2)
|
|
id_ti_0 = builder.add_immediate_transition(priority=0, weight=0.5, name="ti_0")
|
|
id_tt_0 = builder.add_timed_transition(priority=0, rate=0.5, num_servers=2, name="tt_0")
|
|
|
|
gspn = builder.build_gspn()
|
|
|
|
export_file = os.path.join(str(tmpdir), "gspn.pnml")
|
|
|
|
# export gspn to pnml
|
|
gspn.export_gspn_pnml_file(export_file)
|
|
|
|
# import gspn
|
|
gspn_parser = stormpy.gspn.GSPNParser()
|
|
gspn_import = gspn_parser.parse(export_file)
|
|
|
|
# test imported gspn
|
|
assert gspn_import.get_name() == gspn.get_name()
|
|
assert gspn_import.get_number_of_timed_transitions() == gspn.get_number_of_timed_transitions()
|
|
assert gspn_import.get_number_of_immediate_transitions() == gspn.get_number_of_immediate_transitions()
|
|
assert gspn_import.get_number_of_places() == gspn.get_number_of_places()
|
|
|
|
p_0 = gspn_import.get_place(id_p_0)
|
|
assert p_0.get_id() == id_p_0
|
|
p_1 = gspn_import.get_place(id_p_1)
|
|
assert p_1.get_name() == "place_1"
|
|
assert p_1.get_capacity() == 2
|
|
assert p_1.get_number_of_initial_tokens() == 3
|
|
assert p_1.has_restricted_capacity()
|
|
|
|
ti_0 = gspn_import.get_immediate_transition("ti_0")
|
|
assert ti_0.get_id() == id_ti_0
|
|
tt_0 = gspn_import.get_timed_transition("tt_0")
|
|
assert tt_0.get_id() == id_tt_0
|