22
33import enum
44
5+ import numpy as np
6+ from nitypes .scalar import Scalar
7+ from nitypes .waveform import AnalogWaveform
8+
59
610class MyIntFlags (enum .IntFlag ):
711 """Example of an IntFlag enum."""
@@ -29,21 +33,21 @@ class MyStrEnum(str, enum.Enum):
2933
3034all_types_with_values = {
3135 # supported scalar types
32- "bool_scalar " : True ,
33- "bytes_scalar " : b"robotext" ,
34- "float_scalar " : 13.12 ,
35- "int_scalar " : 42 ,
36- "str_scalar " : "sample string" ,
36+ "bool " : True ,
37+ "bytes " : b"robotext" ,
38+ "float " : 13.12 ,
39+ "int " : 42 ,
40+ "str " : "sample string" ,
3741 # supported collection types
3842 "bool_collection" : [True , False , True ],
3943 "bytes_collection" : [b"one" , b"two" , b"three" ],
4044 "float_collection" : [1.1 , 2.2 , 3.3 ],
4145 "int_collection" : [1 , 2 , 3 ],
4246 "str_collection" : ["one" , "two" , "three" ],
4347 # supported enum and flag types
44- "intflags_scalar " : MyIntFlags .VALUE1 | MyIntFlags .VALUE4 ,
45- "intenum_scalar " : MyIntEnum .VALUE20 ,
46- "strenum_scalar " : MyStrEnum .VALUE3 ,
48+ "intflags " : MyIntFlags .VALUE1 | MyIntFlags .VALUE4 ,
49+ "intenum " : MyIntEnum .VALUE20 ,
50+ "strenum " : MyStrEnum .VALUE3 ,
4751 "intflags_collection" : [MyIntFlags .VALUE1 , MyIntFlags .VALUE2 , MyIntFlags .VALUE4 ],
4852 "intenum_collection" : [MyIntEnum .VALUE10 , MyIntEnum .VALUE20 , MyIntEnum .VALUE30 ],
4953 "strenum_collection" : [MyStrEnum .VALUE1 , MyStrEnum .VALUE2 , MyStrEnum .VALUE3 ],
@@ -52,4 +56,7 @@ class MyStrEnum(str, enum.Enum):
5256 "tuple" : (4 , 5 , 6 ),
5357 "set" : {7 , 8 , 9 },
5458 "frozenset" : frozenset ([10 , 11 , 12 ]),
59+ # NI types
60+ "nitypes_Scalar" : Scalar (42 , "m" ),
61+ "nitypes_AnalogWaveform" : AnalogWaveform .from_array_1d (np .array ([1.0 , 2.0 , 3.0 ])),
5562}
0 commit comments