#!/usr/bin/env python3

# License: MIT License


import unittest
import glob
import os
import random

from pyctbn.legacy.utility.json_importer import JsonImporter
from pyctbn.legacy.structure_graph.sample_path import SamplePath
from pyctbn.legacy.structure_graph.trajectory import Trajectory
from pyctbn.legacy.structure_graph.structure import Structure


class TestSamplePath(unittest.TestCase):

    @classmethod
    def setUpClass(cls) -> None:
        cls.read_files = glob.glob(os.path.join('./tests/data', "*.json"))

    def test_init_not_initialized_importer(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        self.assertRaises(RuntimeError, SamplePath, importer)

    def test_init_not_filled_dataframse(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        importer.clear_concatenated_frame()
        self.assertRaises(RuntimeError, SamplePath, importer)

    def test_init(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        s1 = SamplePath(importer)
        self.assertIsNone(s1.trajectories)
        self.assertIsNone(s1.structure)
        self.assertFalse(s1._importer.concatenated_samples.empty)
        self.assertIsNone(s1._total_variables_count)

    def test_build_trajectories(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        s1 = SamplePath(importer)
        s1.build_trajectories()
        self.assertIsInstance(s1.trajectories, Trajectory)

    def test_build_structure(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        s1 = SamplePath(importer)
        s1.build_structure()
        self.assertIsInstance(s1.structure, Structure)
        self.assertEqual(s1._total_variables_count, len(s1._importer.sorter))

    def test_build_structure_bad_sorter(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        s1 = SamplePath(importer)
        importer._sorter[0],importer._sorter[1]= importer._sorter[1],importer._sorter[0]

        self.assertRaises(RuntimeError, s1.build_structure)

    def test_build_saplepath_no_prior_net_structure(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        importer._df_structure = None
        s1 = SamplePath(importer)
        s1.build_trajectories()
        s1.build_structure()
        self.assertFalse(s1.structure.edges)

    def test_buid_samplepath_no_variables(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        importer._df_variables = None
        self.assertRaises(RuntimeError, SamplePath, importer)

    def test_buid_samplepath_no_concatenated_samples(self):
        importer = JsonImporter(self.read_files[2], 'samples', 'dyn.str', 'variables', 'Time', 'Name')
        importer.import_data(0)
        importer._concatenated_samples = None
        self.assertRaises(RuntimeError, SamplePath, importer)

if __name__ == '__main__':
    unittest.main()