Luca Moretti
4 years ago
parent
1d355362b1
commit
6be81aaec2
@ -1,90 +1,67 @@ |
||||
import sys |
||||
sys.path.append("../../classes/") |
||||
|
||||
import unittest |
||||
import numpy as np |
||||
import glob |
||||
import os |
||||
|
||||
import structure_graph.network_graph as ng |
||||
import structure_graph.sample_path as sp |
||||
import structure_graph.set_of_cims as sofc |
||||
import estimators.parameters_estimator as pe |
||||
import utility.json_importer as ji |
||||
from ...classes.structure_graph.network_graph import NetworkGraph |
||||
from ...classes.structure_graph.sample_path import SamplePath |
||||
from ...classes.structure_graph.set_of_cims import SetOfCims |
||||
from ...classes.estimators.parameters_estimator import ParametersEstimator |
||||
from ...classes.utility.json_importer import JsonImporter |
||||
|
||||
|
||||
class TestParametersEstimatior(unittest.TestCase): |
||||
|
||||
@classmethod |
||||
def setUpClass(cls) -> None: |
||||
cls.read_files = glob.glob(os.path.join('../../data', "*.json")) |
||||
cls.importer = ji.JsonImporter("../../data/networks_and_trajectories_ternary_data_01_3.json", 'samples', 'dyn.str', 'variables', 'Time', 'Name') |
||||
cls.s1 = sp.SamplePath(cls.importer) |
||||
cls.read_files = glob.glob(os.path.join('./main_package/data', "*.json")) |
||||
cls.array_indx = 0 |
||||
cls.importer = JsonImporter(cls.read_files[0], 'samples', 'dyn.str', 'variables', 'Time', 'Name') |
||||
cls.importer.import_data(cls.array_indx) |
||||
cls.s1 = SamplePath(cls.importer) |
||||
cls.s1.build_trajectories() |
||||
cls.s1.build_structure() |
||||
cls.g1 = ng.NetworkGraph(cls.s1.structure) |
||||
cls.g1.init_graph() |
||||
print(cls.s1.structure.edges) |
||||
print(cls.s1.structure.nodes_values) |
||||
|
||||
def test_fast_init(self): |
||||
for node in self.g1.nodes: |
||||
g = ng.NetworkGraph(self.s1.structure) |
||||
for node in self.s1.structure.nodes_labels: |
||||
g = NetworkGraph(self.s1.structure) |
||||
g.fast_init(node) |
||||
p1 = pe.ParametersEstimator(self.s1, g) |
||||
self.assertEqual(p1.sample_path, self.s1) |
||||
self.assertEqual(p1.net_graph, g) |
||||
self.assertIsNone(p1.single_set_of_cims) |
||||
p1 = ParametersEstimator(self.s1.trajectories, g) |
||||
self.assertEqual(p1._trajectories, self.s1.trajectories) |
||||
self.assertEqual(p1._net_graph, g) |
||||
self.assertIsNone(p1._single_set_of_cims) |
||||
p1.fast_init(node) |
||||
self.assertIsInstance(p1.single_set_of_cims, sofc.SetOfCims) |
||||
self.assertIsInstance(p1._single_set_of_cims, SetOfCims) |
||||
|
||||
def test_compute_parameters_for_node(self): |
||||
for indx, node in enumerate(self.g1.nodes): |
||||
for indx, node in enumerate(self.s1.structure.nodes_labels): |
||||
print(node) |
||||
g = ng.NetworkGraph(self.s1.structure) |
||||
g = NetworkGraph(self.s1.structure) |
||||
g.fast_init(node) |
||||
p1 = pe.ParametersEstimator(self.s1, g) |
||||
p1 = ParametersEstimator(self.s1.trajectories, g) |
||||
p1.fast_init(node) |
||||
sofc1 = p1.compute_parameters_for_node(node) |
||||
sampled_cims = self.aux_import_sampled_cims('dyn.cims') |
||||
sc = list(sampled_cims.values()) |
||||
print(sampled_cims.values()) |
||||
#print(sc[indx]) |
||||
self.equality_of_cims_of_node(sc[indx], sofc1.actual_cims) |
||||
self.equality_of_cims_of_node(sc[indx], sofc1._actual_cims) |
||||
|
||||
def equality_of_cims_of_node(self, sampled_cims, estimated_cims): |
||||
#print(sampled_cims) |
||||
#print(estimated_cims) |
||||
self.assertEqual(len(sampled_cims), len(estimated_cims)) |
||||
for c1, c2 in zip(sampled_cims, estimated_cims): |
||||
self.cim_equality_test(c1, c2.cim) |
||||
|
||||
def cim_equality_test(self, cim1, cim2): |
||||
for r1, r2 in zip(cim1, cim2): |
||||
self.assertTrue(np.all(np.isclose(r1, r2, 1e-01, 1e-01) == True)) |
||||
self.assertTrue(np.all(np.isclose(r1, r2, 1e-01, 1e-01) is True)) |
||||
|
||||
def aux_import_sampled_cims(self, cims_label): |
||||
i1 = ji.JsonImporter(self.read_files[0], '', '', '', '', '') |
||||
i1 = JsonImporter(self.read_files[0], '', '', '', '', '') |
||||
raw_data = i1.read_json_file() |
||||
return i1.import_sampled_cims(raw_data, 0, cims_label) |
||||
|
||||
""" |
||||
def test_init(self): |
||||
self.aux_test_init(self.s1, self.g1) |
||||
|
||||
def test_init_sets_of_cims_container(self): |
||||
self.aux_test_init_sets_cims_container(self.s1, self.g1) |
||||
|
||||
def aux_test_init(self, sample_p, graph): |
||||
pe1 = pe.ParametersEstimator(sample_p, graph) |
||||
self.assertEqual(sample_p, pe1.sample_path) |
||||
self.assertEqual(graph, pe1.net_graph) |
||||
self.assertIsNone(pe1.sets_of_cims_struct) |
||||
return i1.import_sampled_cims(raw_data, self.array_indx, cims_label) |
||||
|
||||
def aux_test_init_sets_cims_container(self, sample_p, graph): |
||||
pe1 = pe.ParametersEstimator(sample_p, graph) |
||||
pe1.init_sets_cims_container() |
||||
self.assertIsInstance(pe1.sets_of_cims_struct, scc.SetsOfCimsContainer) |
||||
|
||||
def test_compute_parameters(self): |
||||
self.aux_test_compute_parameters(self.s1, self.g1) |
||||
""" |
||||
if __name__ == '__main__': |
||||
unittest.main() |
||||
|
||||
Reference in new issue