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)
