import unittest

from urh.signalprocessing.encoding import encoding

class TestWhitening(unittest.TestCase):
    # Testmethode muss immer mit Präfix test_* starten
    def test_whitening(self):
        e = encoding()

        # Test 1
        e.data_whitening_sync = e.hex2bit("67686768")
        original_inpt = e.hex2bit("aaaaaaaa67686768f9ca03909567ba76a8") + [False] # Korrektes Signal, bitgenau
        inpt = original_inpt.copy()
        #print (e.bit2hex(inpt))
        output, err = e.apply_data_whitening(True, inpt)
        #print (e.bit2hex(output), err)
        newinpt, err = e.apply_data_whitening(False, output)
        #print (e.bit2hex(newinpt), newinpt, err)
        self.assertEqual(original_inpt, newinpt)