import os
import unittest
from commons.pyRepetUnit.fastaTranslation.allFrames.TranslateInAllFramesAndReplaceStopByX import TranslateInAllFramesAndReplaceStopByX
from commons.core.seq.Bioseq import Bioseq
from commons.core.utils.FileUtils import FileUtils

class Test_F_TranslateInAllFramesAndReplaceStopByX(unittest.TestCase):
    
    def setUp(self):
        self.bioseq = Bioseq()
        self.preProcess = TranslateInAllFramesAndReplaceStopByX()
        self._inputFile = "./datas/test_input_nt.fa" 
        self._outputFile =  "./datas/PreProcessResult.fa" 
        
        
    def testAcceptanceTranslateAfastaFileInAllFrame(self):
        self.preProcess.setInputFile(self._inputFile)
        self.preProcess.setOutputFile(self._outputFile)
        self.preProcess.run()
        self.assertTrue(FileUtils.are2FilesIdentical(self._outputFile, "./datas/test_input_aa.fa"))
        os.system("rm " + self._outputFile)
        
if __name__ == "__main__":                 
    unittest.main()