15 lines
449 B
Python
15 lines
449 B
Python
import unittest
|
|
from majority_element import majority
|
|
|
|
class TestMajority(unittest.TestCase):
|
|
def test_basic(self):
|
|
self.assertEqual(majority([1,1,2]), 1)
|
|
self.assertEqual(majority([2,2,1,1,1]), 1)
|
|
self.assertIsNone(majority([1,2,3]))
|
|
self.assertIsNone(majority([]))
|
|
self.assertEqual(majority([5]), 5)
|
|
self.assertEqual(majority([1,1,1,2,2,2,2]), 2)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|