import pytest

from typing import List, Optional, Union
from types import NoneType
from nwb_linkml.annotations import get_inner_types


@pytest.mark.parametrize(("annotation", "inner_types"), [(List[str | None], (str, NoneType))])
def test_get_inner_types(annotation, inner_types):
    got_inner_types = get_inner_types(annotation)
    assert len(got_inner_types) == len(inner_types)
    for got, expected in zip(got_inner_types, inner_types):
        assert got is expected