Processing_unittest.py

import unittest
from Processing import process_main

class ProcessingTest(unittest.TestCase):
    def testCycle(self):
        self.assertEqual(process_main(['doiz1'],1,1,True), (['doiz1', 'doiz2'], [['doiz2', 'doiz1'], ['doiz1', 'doiz2']]))
        self.assertEqual(process_main(['doiz1'],2,2,True), (['doiz1', 'doiz2'], [['doiz2', 'doiz1'], ['doiz1', 'doiz2']]))

    #def testBigCycle(self):

    #def testEmptyHeight(self):

    #def testEmptyDepth(self):

    def testEmptyDepthHeight(self):
        self.assertEqual(process_main(['doi1'],0,0,True), (['doi1'], []))
        self.assertEqual(process_main(['doi1', 'doi2'],0,0,True), (['doi1','doi2'], [['doi1', 'doi2']]))
        self.assertEqual(process_main(['doi1', 'doi2', 'doi3'],0,0,True), (['doi1','doi2', 'doi3'], [['doi3', 'doi1'], ['doi1', 'doi2']]))

    #def testInnerEdges(self):

if __name__ == "__main__":
    unittest.main()