text: from algorithms.heap import (
      BinaryHeap,
      get_skyline,
      max_sliding_window,
      k_closest_points
  )
  
  import unittest
  
  
  class TestBinaryHeap(unittest.TestCase):
      """
          Test suite for the binary_heap data structures
      """
  
      def setUp(self):
          self.min_heap = BinaryHeap()
          self.min_heap.insert(4)
          self.min_heap.insert(50)
          self.min_heap.insert(7)
          self.min_heap.insert(55)
          self.min_heap.insert(90)
          self.min_heap.insert(87)
  
      def test_insert(self):
          # Before insert 2: [0, 4, 50, 7, 55, 90, 87]
          # After insert:  [0, 2, 50, 4, 55, 90, 87, 7]
          self.min_heap.insert(2)
          self.assertEqual([0, 2, 50, 4, 55, 90, 87, 7],
                           self.min_heap.heap)
          self.assertEqual(7, self.min_heap.currentSize)
  
      def test_remove_min(self):
          ret = self.min_heap.remove_min()
          # Before remove_min : [0, 4, 50, 7, 55, 90, 87]
          # After remove_min: [7, 50, 87, 55, 90]
          # Test return value
          self.assertEqual(4, ret)
          self.assertEqual([0, 7, 50, 87, 55, 90],
                           self.min_heap.heap)
          self.assertEqual(5, self.min_heap.currentSize)
  
  
  class TestSuite(unittest.TestCase):
      def test_get_skyline(self):
          buildings = [[2, 9, 10], [3, 7, 15], [5, 12, 12],
                       [15, 20, 10], [19, 24, 8]]
          # Expect output
          output = [[2, 10], [3, 15], [7, 12], [12, 0], [15, 10],
                    [20, 8], [24, 0]]
          self.assertEqual(output, get_skyline(buildings))
  
      def test_max_sliding_window(self):
          nums = [1, 3, -1, -3, 5, 3, 6, 7]
          self.assertEqual([3, 3, 5, 5, 6, 7], max_sliding_window(nums, 3))
  
      def test_k_closest_points(self):
          points = [(1, 0), (2, 3), (5, 2), (1, 1), (2, 8), (10, 2), (-1, 0), (-2, -2)]
          self.assertEqual([(-1, 0), (1, 0)], k_closest_points.k_closest(points, 2))
          self.assertEqual([(1, 1), (-1, 0), (1, 0)], k_closest_points.k_closest(points, 3))
          self.assertEqual([(-2, -2), (1, 1), (1, 0), (-1, 0)], k_closest_points.k_closest(points, 4))
          self.assertEqual([(10, 2), (2, 8), (5, 2), (-2, -2), (2, 3),
                            (1, 0), (-1, 0), (1, 1)], k_closest_points.k_closest(points, 8))
  
  
  if __name__ == "__main__":
      unittest.main()
  