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.
124 lines
3.9 KiB
124 lines
3.9 KiB
import os
|
|
|
|
import stormpy
|
|
|
|
from configurations import gspn
|
|
|
|
from helpers.helper import get_example_path
|
|
|
|
|
|
@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 place.has_restricted_capacity() == False
|
|
# todo this does not work (boost::optional<uint64_t> ?):
|
|
place.set_capacity(cap = 5)
|
|
# assert place.has_restricted_capacity() == True
|
|
# 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 TimedTrasnition
|
|
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()
|
|
|
|
# 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() == True
|
|
ti_weight = 0.2
|
|
ti.set_weight(ti_weight)
|
|
assert ti_weight == ti.get_weight()
|
|
assert ti.no_weight_attached() == False
|
|
|
|
#def test_gspn(self):
|
|
# todo
|
|
#immediateTransitions =
|
|
#gspn = stormpy.gspn.GSPN("gspn", places, immediateTransitions, timedTransitions, orderedPartitions, actualExprManager, constantsSubstitution)
|
|
|
|
def test_build_gspn(self):
|
|
|
|
gspn_name = "gspn_test"
|
|
builder = stormpy.gspn.GSPNBuilder()
|
|
builder.set_name(gspn_name)
|
|
# todo place tests, set_place_layout_info (boost)
|
|
#p_id_0 = builder.add_place()
|
|
# p_id_0 = builder.add_place(capacity = 1, initialTokens = 0, name = "place_test")
|
|
# p_layout = stormpy.gspn.LayoutInfo(1, 2)
|
|
# builder.set_place_layout_info(p_id_0, t_layout)
|
|
|
|
ti_id_0 = builder.add_immediate_transition()
|
|
ti_id_1 = builder.add_immediate_transition(priority=0, weight=0.5, name="ti_1")
|
|
|
|
tt_id_0 = builder.add_timed_transition(priority=0, rate=0.5, name="tt_0")
|
|
# todo problems with sumServers (boost)
|
|
# tt_id_1 = builder.add_timed_transition(priority=0, rate=0.5, numServers=2, name="tt_1")
|
|
|
|
# todo tests for add_ (add_place needed)
|
|
# builder.add_input_arc(ti_id_0, ti_id_1, multiplicity = 2)
|
|
# todo test addNormalArc
|
|
|
|
t_layout = stormpy.gspn.LayoutInfo(1, 2)
|
|
builder.set_transition_layout_info(ti_id_0, t_layout)
|
|
|
|
gspn = builder.build_gspn()
|
|
assert gspn.name() == gspn_name
|
|
|
|
gspn_new_name = "new_name"
|
|
gspn.set_name(gspn_new_name)
|
|
assert gspn.name() == gspn_new_name
|
|
|
|
def test_export_to_pnpro(self, tmpdir):
|
|
gspn_name = "gspn_test"
|
|
builder = stormpy.gspn.GSPNBuilder()
|
|
builder.set_name(gspn_name)
|
|
gspn = builder.build_gspn()
|
|
# todo add trans and places (layout?)
|
|
assert gspn.name() == gspn_name
|
|
#todo assert gspn.nr trans etc
|
|
export_file = os.path.join(str(tmpdir), "gspn.pnpro")
|
|
gspn.export_gspn_pnpro_file(export_file)
|
|
|
|
|
|
#todo: load and assert gspn.nr trans etc
|
|
# gspn_from_file = stormpy.gspn.load_gspn_file(export_file)
|
|
# assert gspn_from_file.nr ==...
|
|
|
|
#def test_export_to_pnml(self, tmpdir):
|
|
|
|
|
|
|
|
|