research/argus
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests(pyargus): add test cases for signals

Browse source
This commit is contained in:
Anand Balakrishnan 2023-08-31 17:11:52 -07:00
parent ab0a2c6d85
commit 137c22cd70
11 changed files with 193 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

13
pyargus/argus/__init__.pyi Normal file
View file

@ -0,0 +1,13 @@
from typing import List, Optional, Tuple, Type, Union
from argus._argus import DType as DType
from argus.exprs import ConstBool, ConstFloat, ConstInt, ConstUInt, VarBool, VarFloat, VarInt, VarUInt
from argus.signals import Signal
AllowedDtype = Union[bool, int, float]
def declare_var(name: str, dtype: Union[DType, Type[AllowedDtype]]) -> Union[VarBool, VarInt, VarUInt, VarFloat]: ...
def literal(value: AllowedDtype) -> Union[ConstBool, ConstInt, ConstUInt, ConstFloat]: ...
def signal(
dtype: Union[DType, Type[AllowedDtype]], *, data: Optional[Union[AllowedDtype, List[Tuple[float, AllowedDtype]]]] = ...
) -> Signal: ...