import pytest
from solution import Solution


@pytest.mark.parametrize(
    "nums1, nums2, expected",
    [
        ([1, 3], [2], 2),
        ([1, 2], [3, 4], 2.5),
    ],
)
def test_median_sorted(nums1, nums2, expected):
    assert Solution().findMedianSortedArrays(nums1, nums2) == expected