From 56a5af4f68ca2be6f926f4abeef6cc58c4b63536 Mon Sep 17 00:00:00 2001
From: Luca Moretti <lucamoretti96@gmail.com>
Date: Tue, 22 Dec 2020 13:53:01 +0100
Subject: [PATCH] Added file for test on server

---
 .../test_structure_score_based_estimator.py   | 59 ++++++++++++++
 ..._structure_score_based_estimator_server.py | 80 +++++++++++++++++++
 2 files changed, 139 insertions(+)
 create mode 100644 main_package/copia/test_structure_score_based_estimator.py
 create mode 100644 main_package/tests/estimators/test_structure_score_based_estimator_server.py

diff --git a/main_package/copia/test_structure_score_based_estimator.py b/main_package/copia/test_structure_score_based_estimator.py
new file mode 100644
index 0000000..e0ec2e5
--- /dev/null
+++ b/main_package/copia/test_structure_score_based_estimator.py
@@ -0,0 +1,59 @@
+import sys
+sys.path.append("../../classes/")
+import glob
+import math
+import os
+import unittest
+
+import networkx as nx
+import numpy as np
+import psutil
+from line_profiler import LineProfiler
+import copy
+
+import utility.cache as ch
+import structure_graph.sample_path as sp
+import estimators.structure_score_based_estimator as se
+import utility.json_importer as ji
+
+from multiprocessing import set_start_method
+
+
+
+
+class TestStructureScoreBasedEstimator(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        #cls.read_files = glob.glob(os.path.join('../../data', "*.json"))
+        cls.importer = ji.JsonImporter("/home/alessandro/Documents/ctbn_cba/data/networks_and_trajectories_ternary_data_15.json", 'samples', 'dyn.str', 'variables', 'Time', 'Name')
+        cls.s1 = sp.SamplePath(cls.importer)
+        cls.s1.build_trajectories()
+        cls.s1.build_structure()
+
+
+
+    def test_structure(self):
+        true_edges = copy.deepcopy(self.s1.structure.edges)
+        true_edges = set(map(tuple, true_edges))
+
+        set_start_method("spawn")
+        se1 = se.StructureScoreBasedEstimator(self.s1)
+        edges = se1.estimate_structure(
+                                    max_parents = None,
+                                    iterations_number = 100,
+                                    patience = 35,
+                                    tabu_length = 15,
+                                    tabu_rules_duration = 15,
+                                    optimizer = 'tabu',
+                                    disable_multiprocessing=False
+                                    )
+        
+
+        self.assertEqual(edges, true_edges)
+        
+
+
+if __name__ == '__main__':
+    unittest.main()
+
diff --git a/main_package/tests/estimators/test_structure_score_based_estimator_server.py b/main_package/tests/estimators/test_structure_score_based_estimator_server.py
new file mode 100644
index 0000000..813d981
--- /dev/null
+++ b/main_package/tests/estimators/test_structure_score_based_estimator_server.py
@@ -0,0 +1,80 @@
+import sys
+sys.path.append("../../classes/")
+import glob
+import math
+import os
+import unittest
+
+import networkx as nx
+import numpy as np
+import psutil
+from line_profiler import LineProfiler
+import copy
+
+import utility.cache as ch
+import structure_graph.sample_path as sp
+import estimators.structure_score_based_estimator as se
+import utility.json_importer as ji
+
+
+
+class TestStructureScoreBasedEstimator(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        pass
+
+
+
+    def test_structure(self):
+        #cls.read_files = glob.glob(os.path.join('../../data', "*.json"))
+        self.importer = ji.JsonImporter("/home/alessandro/Documents/ctbn_cba/data/networks_and_trajectories_ternary_data_6.json", 'samples', 'dyn.str', 'variables', 'Time', 'Name')
+        self.s1 = sp.SamplePath(cls.importer)
+        self.s1.build_trajectories()
+        self.s1.build_structure()
+
+        true_edges = copy.deepcopy(self.s1.structure.edges)
+        true_edges = set(map(tuple, true_edges))
+
+        
+        se1 = se.StructureScoreBasedEstimator(self.s1)
+        edges = se1.estimate_structure(
+                                    max_parents = None,
+                                    iterations_number = 100,
+                                    patience = 35,
+                                    tabu_length = 15,
+                                    tabu_rules_duration = 15,
+                                    optimizer = 'tabu',
+                                    disable_multiprocessing=False
+                                    )
+
+
+        self.importer = ji.JsonImporter("/home/alessandro/Documents/ctbn_cba/data/networks_and_trajectories_ternary_data_6.json", 'samples', 'dyn.str', 'variables', 'Time', 'Name')
+        self.s1 = sp.SamplePath(cls.importer)
+        self.s1.build_trajectories()
+        self.s1.build_structure()
+
+        true_edges = copy.deepcopy(self.s1.structure.edges)
+        true_edges = set(map(tuple, true_edges))
+
+        
+        se1 = se.StructureScoreBasedEstimator(self.s1)
+        edges = se1.estimate_structure(
+                                    max_parents = None,
+                                    iterations_number = 100,
+                                    patience = 35,
+                                    tabu_length = 15,
+                                    tabu_rules_duration = 15,
+                                    optimizer = 'tabu',
+                                    disable_multiprocessing=True
+                                    )
+        
+        
+
+        self.assertEqual(edges, true_edges)
+        
+
+
+if __name__ == '__main__':
+    unittest.main()
+