#include "s0004_median_of_two_sorted_arrays.hpp"

#include <gtest/gtest.h>

TEST(Problem4, Case1) {
  std::vector<int> nums1{1, 3};
  std::vector<int> nums2{2};
  double answer{2};
  S0004 solution;
  EXPECT_EQ(solution.findMedianSortedArrays(nums1, nums2), answer);
}

TEST(Problem4, Case2) {
  std::vector<int> nums1{1, 2};
  std::vector<int> nums2{3, 4};
  double answer{2.5};
  S0004 solution;
  EXPECT_EQ(solution.findMedianSortedArrays(nums1, nums2), answer);
}