Old engine for Continuous Time Bayesian Networks. Superseded by reCTBN. 🐍
https://github.com/madlabunimib/PyCTBN
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.
25 lines
664 B
25 lines
664 B
4 years ago
|
import unittest
|
||
|
|
|
||
|
|
import sets_of_cims_container as scc
|
||
|
|
import set_of_cims as sc
|
||
|
|
|
||
|
|
|
||
|
|
class TestSetsOfCimsContainer(unittest.TestCase):
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls) -> None:
|
||
|
|
cls.variables = ['X', 'Y', 'Z']
|
||
|
|
cls.states_per_node = [3, 3, 3]
|
||
|
|
cls.parents_states_list = [[], [3], [3, 3]]
|
||
|
|
|
||
|
|
def test_init(self):
|
||
|
|
c1 = scc.SetsOfCimsContainer(self.variables, self.states_per_node, self.parents_states_list)
|
||
|
|
self.assertEqual(len(c1.sets_of_cims), len(self.variables))
|
||
|
|
for set_of_cims in c1.sets_of_cims:
|
||
|
|
self.assertIsInstance(set_of_cims, sc.SetOfCims)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|