''' Created on Oct 31, 2018 @author: haidong ''' import unittest from sources.majority_element import getMajorityElement class Test(unittest.TestCase): def testGetMajorityElement1(self): a = [1,1] self.assertEqual(1, getMajorityElement(a, 0, len(a)-1)) def testGetMajorityElement2(self): a = [1,2] self.assertEqual(-1, getMajorityElement(a, 0, len(a)-1)) def testGetMajorityElement3(self): a = [1,2,1] self.assertEqual(1, getMajorityElement(a, 0, len(a)-1)) def testGetMajorityElement4(self): a = [1,2,2] self.assertEqual(2, getMajorityElement(a, 0, len(a)-1)) def testGetMajorityElement5(self): a = [1] self.assertEqual(1, getMajorityElement(a, 0, len(a)-1)) def testGetMajorityElement6(self): a = [1, 2, 2, 2] self.assertEqual(2, getMajorityElement(a, 0, len(a)-1)) def testGetMajorityElement7(self): a = [2, 3, 9, 2, 2] self.assertEqual(2, getMajorityElement(a, 0, len(a)-1)) if __name__ == "__main__": #import sys;sys.argv = ['', 'Test.testName'] unittest.main()