def test_first(self): obj = MedianFinder() obj.addNum(1) obj.addNum(2) self.assertEqual(obj.findMedian(), 1.5)
from median import MedianFinder import numpy as np obj = MedianFinder() for i in np.random.randint(-10, 10, 7): obj.addNum(i) median = obj.findMedian() print('#---------------------------------') print(f'Median :: {median}\nFull array :: {obj.data}\n')
def test_second(self): obj = MedianFinder() obj.addNum(1) obj.addNum(2) obj.addNum(3) self.assertEqual(obj.findMedian(), 2)