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

1from collections.abc import Callable 

2from typing import TypeVar, overload 

3 

4TNum = TypeVar('TNum', int, float) 

5 

6 

7@overload 

8def add(a: int, /) -> Callable[[TNum], TNum]: ... 

9 

10 

11@overload 

12def add(a: float, /) -> Callable[[int | float], float]: ... 

13 

14 

15@overload 

16def add(a: int, b: TNum, /) -> TNum: ... 

17 

18 

19@overload 

20def add(a: TNum, b: int, /) -> TNum: ... 

21# rename to value, addend 

22def add( 

23 a: int | float, 

24 b: int | float | None = None, 

25 /, 

26) -> int | float | Callable[[int | float], int | float] | Callable[[TNum], TNum]: 

27 if b is None: 

28 return lambda b: a + b 

29 return a + b