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.
32 lines
996 B
32 lines
996 B
import unittest
|
|
import glob
|
|
import os
|
|
import json_importer as ji
|
|
import sample_path as sp
|
|
import trajectory as tr
|
|
import structure as st
|
|
|
|
|
|
class TestSamplePath(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
cls.read_files = glob.glob(os.path.join('../data', "*.json"))
|
|
cls.importer = ji.JsonImporter(cls.read_files[0], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
|
|
|
|
def test_init(self):
|
|
s1 = sp.SamplePath(self.importer)
|
|
s1.build_trajectories()
|
|
self.assertIsNotNone(s1.trajectories)
|
|
self.assertIsInstance(s1.trajectories, tr.Trajectory)
|
|
s1.build_structure()
|
|
self.assertIsNotNone(s1.structure)
|
|
self.assertIsInstance(s1.structure, st.Structure)
|
|
self.assertTrue(s1.importer.concatenated_samples.empty)
|
|
self.assertEqual(s1.total_variables_count, len(s1.importer.sorter))
|
|
print(s1.structure)
|
|
print(s1.trajectories)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|