def test_matrix_change_state_block_method(self):
Q = ciw.Simulation(ciw.create_network_from_yml(
'ciw/tests/testing_parameters/params.yml'))
B = ciw.MatrixTracker(Q)
B.state = [[[[], [], [], []],
[[], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[2, 3, 1, 0]]
B.change_state_block(1, 3, 2)
self.assertEqual(B.state, [[[[], [], [1], []],
[[], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[2, 3, 1, 0]])
B.change_state_block(2, 1, 0)
self.assertEqual(B.state, [[[[], [], [1], []],
[[2], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[2, 3, 1, 0]])
B.change_state_block(1, 3, 0)
self.assertEqual(B.state, [[[[], [], [1, 3], []],
[[2], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[2, 3, 1, 0]])
def test_matrix_init_method(self):
Q = ciw.Simulation(
ciw.load_parameters(
'ciw/tests/datafortesting/logs_test_for_simulation/parameters.yml'
))
B = ciw.MatrixTracker(Q)
self.assertEqual(B.simulation, Q)
self.assertEqual(B.state,
[[[[], [], [], []], [[], [], [], []],
[[], [], [], []], [[], [], [], []]], [0, 0, 0, 0]])
def test_matrix_init_method(self):
Q = ciw.Simulation(ciw.create_network_from_yml(
'ciw/tests/testing_parameters/params.yml'))
B = ciw.MatrixTracker(Q)
self.assertEqual(B.simulation, Q)
self.assertEqual(B.state, [[[[], [], [], []],
[[], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[0, 0, 0, 0]])
def test_matrix_hash_state_method(self):
Q = ciw.Simulation(
ciw.load_parameters(
'ciw/tests/datafortesting/logs_test_for_simulation/parameters.yml'
))
B = ciw.MatrixTracker(Q)
B.state = [[[[], [], [1, 3], []], [[2], [], [], []], [[], [], [], []],
[[], [], [], []]], [2, 3, 0, 0]]
self.assertEqual(B.hash_state(),
((((), (), (1, 3), ()), ((2, ), (), (), ()),
((), (), (), ()), ((), (), (), ())), (2, 3, 0, 0)))
def test_matrix_hash_state_method(self):
Q = ciw.Simulation(ciw.create_network_from_yml(
'ciw/tests/testing_parameters/params.yml'))
B = ciw.MatrixTracker(Q)
B.state = [[[[], [], [1, 3], []],
[[2], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[2, 3, 0, 0]]
self.assertEqual(B.hash_state(), ((((), (), (1, 3), ()),
((2,), (), (), ()),
((), (), (), ()),
((), (), (), ())),
(2, 3, 0, 0)))
def test_matrix_change_state_accept_method(self):
Q = ciw.Simulation(ciw.create_network_from_yml(
'ciw/tests/testing_parameters/params.yml'))
B = ciw.MatrixTracker(Q)
self.assertEqual(B.state, [[[[], [], [], []],
[[], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[0, 0, 0, 0]])
B.change_state_accept(1, 1)
self.assertEqual(B.state, [[[[], [], [], []],
[[], [], [], []],
[[], [], [], []],
[[], [], [], []]],
[1, 0, 0, 0]])
def test_matrix_change_state_release_method(self):
Q = ciw.Simulation(
ciw.load_parameters(
'ciw/tests/datafortesting/logs_test_for_simulation/parameters.yml'
))
B = ciw.MatrixTracker(Q)
B.state = [[[[], [], [1, 3], []], [[2], [], [], []], [[], [], [], []],
[[], [], [], []]], [2, 3, 1, 0]]
B.change_state_release(3, 1, 2, False)
self.assertEqual(B.state,
[[[[], [], [1, 3], []], [[2], [], [], []],
[[], [], [], []], [[], [], [], []]], [2, 3, 0, 0]])
B.change_state_release(1, 3, 0, True)
self.assertEqual(B.state,
[[[[], [], [2], []], [[1], [], [], []],
[[], [], [], []], [[], [], [], []]], [1, 3, 0, 0]])