nwb-linkml/nwb_linkml/tests/test_annotations.py
2023-09-26 19:50:49 -07:00

17 lines
No EOL
513 B
Python

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