Coverage for src / remedapy / only.py: 100%

1from collections.abc import Callable, Sequence 

2from typing import TypeVar, overload 

3 

4T = TypeVar('T') 

5 

6 

7@overload 

8def only() -> Callable[[Sequence[T]], T | None]: ... 

9 

10 

11@overload 

12def only(array: Sequence[T], /) -> T | None: ... 

13 

14 

15def only(array: Sequence[T] | None = None, /) -> T | None | Callable[[Sequence[T]], T | None]: 

16 if array is None: 

17 return only 

18 return array[0] if len(array) == 1 else None