import unittest
from src.Sommets import *
from src.Arbre import *
from src.fonctions.encode import *
from src.fonctions.decode import *
from src.fonctions.occurence import *

# pour lancer les tests utilisez:
# python -m unittest discover

class TestUtils(unittest.TestCase):
    def test_Arbre_id(self):
        a = Sommets(10, 'a')
        b = Sommets(8,'b')
        r1 = Sommets(18,'',left=b,right=a)

        arbre1 = Arbre(r1)

        # vérification affectation d'un identifiant unique en créant l'arbre
        liste_id = []
        verification_id = True
        for elem in arbre1.sommets:
            liste_id.append(elem.identifiant)
        if len(liste_id) > len(set(liste_id)):
            verification_id  = False

        self.assertTrue(verification_id)

    def test_Arbre_fusion(self):
        # fusion des arbres
        # on doit retrouver les éléments des 2 arbres + une nouvelle racine
        # les identifiants doivent toujours être uniques
        a = Sommets(10, 'a')
        b = Sommets(8,'b')
        r1 = Sommets(18,'',left=b,right=a)
        c = Sommets(15, 'c')
        d = Sommets(20,'d')
        r2 = Sommets(18,'',left=d,right=c)

        arbre1 = Arbre(r1)
        arbre2 = Arbre(r2)

        l1 = len(arbre1.sommets)
        l2 = len(arbre2.sommets)
        arbre1 += arbre2
        verification_fusion = True
        if l1+l2+1 != len(arbre1.sommets):
            verification_fusion = False

        liste_id2 = []
        verification_id2 = True
        for elem in arbre1.sommets:
            liste_id2.append(elem.identifiant)
        if len(liste_id2) > len(set(liste_id2)):
            verification_id2  = False

        self.assertTrue(verification_fusion)
        self.assertTrue(verification_id2)

    def test_Arbre_recherche(self):
        # recherche de sommet
        a = Sommets(10, 'a')
        b = Sommets(8,'b')
        r1 = Sommets(18,'',left=b,right=a)

        arbre1 = Arbre(r1)

        self.assertEqual(arbre1.recherche(a),a)
        self.assertNotEqual(arbre1.recherche(b),a)

    def test_Arbre_suppression(self):
        a = Sommets(10, 'a')
        b = Sommets(8,'b')
        r1 = Sommets(18,'',left=b,right=a)
        r2 = Sommets(18,'',left=None,right=r1)

        arbre1 = Arbre(r2)

        arbre1 -= r1
        self.assertEqual(arbre1.recherche(a),None)
        self.assertEqual(arbre1.recherche(b),None)
        self.assertEqual(arbre1.recherche(r1),None)
        self.assertEqual(arbre1.recherche(r2),r2)

    def test_occurences(self):
        o1 = calcul_occurence('aaabcc')
        o2 = calcul_occurence('bacaac')
        # dans les 2 cas on doit avoir le même dictionnaire
        # 3 pour a, 2 pour c, 1 pour b
        self.assertEqual(o1,o2)
        self.assertEqual(o1['c'],2)
        self.assertEqual(o1['b'],1)
        self.assertEqual(o1['a'],3)


    def test_encodage_huffman(self):
        string = 'mouton'
        string2 = 'vache'
        (encodedOutput, racine, huffmanEncoding) = huffman_encode(string)
        (encodedOutput2, racine2, huffmanEncoding2) = huffman_encode(string2)
        # l'encodage doit être différent (les dictionnaires et arbres aussi)
        self.assertNotEqual(huffmanEncoding, huffmanEncoding2)
        self.assertNotEqual(encodedOutput, encodedOutput2)
        self.assertNotEqual(Arbre(racine),Arbre(racine2))

    def test_decodage_huffman(self):
        string = 'chèvre'
        (encodedOutput, racine, huffmanEncoding) = huffman_encode(string)
        # on doit être capable de décoder avec la racine de l'arbre ou avec le dictionnaire
        self.assertEqual(string,huffman_decode(encodedOutput,racine))
        self.assertEqual(string,decode_from_dico(encodedOutput,huffmanEncoding))

        # on doit être capable de détecter si le dictionnaire/arbre n'est  pas celui correspondant à un texte encodé
        string2 = 'poule'
        (encodedOutput2, racine2, huffmanEncoding2) = huffman_encode(string2)
        with self.assertRaises(ValueError):
            decode_from_dico(encodedOutput2,huffmanEncoding)
        with self.assertRaises(ValueError):
            huffman_decode(encodedOutput2,racine)


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