test.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

'''
    Class for the testing suite :
	- get the list of all test files
	- create a copy of them on start
	- remove the copy on end
'''

import shutil
import glob
import tempfile
import unittest
import mat

FILE_LIST = zip(glob.glob('clean*'), glob.glob('dirty*'))

class MATTest(unittest.TestCase):
    def setUp(self):
	'''create working copy of the clean and the dirty file in the TMP dir'''
	self.file_list = []
	self.tmpdir = tempfile.mkdtemp()

	for clean, dirty in FILE_LIST:
	    shutil.copy2(clean, self.tmpdir + clean)
	    shutil.copy2(dirty, self.tmpdir + dirty)
	    self.file_list.append((self.tmpdir + clean, self.tmpdir + dirty))

    def tearDown(self):
	'''Remove the tmp folder'''
	shutil.rmtree(self.tmpdir)