BaseTools: Refactor python except statements

[mirror_edk2.git] / BaseTools / Tests / TianoCompress.py

## @file\r
# Unit tests for TianoCompress utility\r
#\r
#  Copyright (c) 2008, Intel Corporation. All rights reserved.<BR>\r
#\r
#  This program and the accompanying materials\r
#  are licensed and made available under the terms and conditions of the BSD License\r
#  which accompanies this distribution.  The full text of the license may be found at\r
#  http://opensource.org/licenses/bsd-license.php\r
#\r
#  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
#  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
#\r
\r
##\r
# Import Modules\r
#\r
import os\r
import random\r
import sys\r
import unittest\r
\r
import TestTools\r
\r
class Tests(TestTools.BaseToolsTest):\r
\r
    def setUp(self):\r
        TestTools.BaseToolsTest.setUp(self)\r
        self.toolName = 'TianoCompress'\r
\r
    def testHelp(self):\r
        result = self.RunTool('--help', logFile='help')\r
        #self.DisplayFile('help')\r
        self.assertTrue(result == 0)\r
\r
    def compressionTestCycle(self, data):\r
        path = self.GetTmpFilePath('input')\r
        self.WriteTmpFile('input', data)\r
        result = self.RunTool(\r
            '-e',\r
            '-o', self.GetTmpFilePath('output1'),\r
            self.GetTmpFilePath('input')\r
            )\r
        self.assertTrue(result == 0)\r
        result = self.RunTool(\r
            '-d',\r
            '-o', self.GetTmpFilePath('output2'),\r
            self.GetTmpFilePath('output1')\r
            )\r
        self.assertTrue(result == 0)\r
        start = self.ReadTmpFile('input')\r
        finish = self.ReadTmpFile('output2')\r
        startEqualsFinish = start == finish\r
        if not startEqualsFinish:\r
            print\r
            print 'Original data did not match decompress(compress(data))'\r
            self.DisplayBinaryData('original data', start)\r
            self.DisplayBinaryData('after compression', self.ReadTmpFile('output1'))\r
            self.DisplayBinaryData('after decomression', finish)\r
        self.assertTrue(startEqualsFinish)\r
\r
    def testRandomDataCycles(self):\r
        for i in range(8):\r
            data = self.GetRandomString(1024, 2048)\r
            self.compressionTestCycle(data)\r
            self.CleanUpTmpDir()\r
\r
TheTestSuite = TestTools.MakeTheTestSuite(locals())\r
\r
if __name__ == '__main__':\r
    allTests = TheTestSuite()\r
    unittest.TextTestRunner().run(allTests)\r
\r
\r