def test_list_output(self):
''' Test list output '''
listOut = iohandle.handleout([])
listOut.add(self.data1)
listOut.add(self.data2)
listOut.add(self.data3)
self.assertEqual(
listOut.close(),
self.datalist
)
def test_pipe_output(self):
''' Test pipe output '''
pipeRecv, pipeSend = multiprocessing.Pipe(False)
pipeOut = iohandle.handleout(pipeSend)
pipeOut.add(self.data1)
pipeOut.add(self.data2)
pipeOut.add(self.data3)
self.assertEqual(
[pipeRecv.recv(),pipeRecv.recv(),pipeRecv.recv()],
self.datalist
)
def test_file_output(self):
''' Test file output '''
fileOut = iohandle.handleout(self.testOut)
fileOut.add(self.data1)
fileOut.add(self.data2)
fileOut.add(self.data3)
fileOut.close()
self.assertTrue(
filecmp.cmp(
self.testOut,
self.testIn
)
)
def test_gzfile_output(self):
''' Test gzip file output '''
fileOut = iohandle.handleout(self.testOutGZ)
fileOut.add(self.data1)
fileOut.add(self.data2)
fileOut.add(self.data3)
fileOut.close()
# Extrcat data from file
fileIn = gzip.open(self.testOutGZ, 'r')
fileContents = fileIn.readlines()
fileIn.close()
# Check file is as expected
self.assertEqual(
fileContents,
[
self.data1 + '\n',
'\t'.join(self.data2) + '\n',
'\t'.join(map(str,self.data3)) + '\n'
]
)
def test_file_failure(self):
''' Test file data failure '''
fileOut = iohandle.handleout(self.testOut)
with self.assertRaises(TypeError):
fileOut.add(['a','b'],['c','d'])
fileOut.close()
def test_type_failure(self):
''' Test output type failure '''
with self.assertRaises(TypeError):
testOut = iohandle.handleout(3)
