55import pathlib
66import sys
77import time
8- from typing import List
8+ from typing import Iterable , List
99
1010import click
1111import grpc
3434
3535
3636@measurement_service .register_measurement
37- @measurement_service .configuration ("pin_name " , nims .DataType .Pin , "Pin1" )
37+ @measurement_service .configuration ("pin_names " , nims .DataType .PinArray1D , [ "Pin1" ] )
3838@measurement_service .configuration ("voltage_level" , nims .DataType .Double , 6.0 )
3939@measurement_service .configuration ("voltage_level_range" , nims .DataType .Double , 6.0 )
4040@measurement_service .configuration ("current_limit" , nims .DataType .Double , 0.01 )
4343@measurement_service .output ("voltage_measurement" , nims .DataType .Double )
4444@measurement_service .output ("current_measurement" , nims .DataType .Double )
4545def measure (
46- pin_name ,
47- voltage_level ,
48- voltage_level_range ,
49- current_limit ,
50- current_limit_range ,
51- source_delay ,
46+ pin_names : Iterable [ str ] ,
47+ voltage_level : float ,
48+ voltage_level_range : float ,
49+ current_limit : float ,
50+ current_limit_range : float ,
51+ source_delay : float ,
5252):
5353 """Source and measure a DC voltage with an NI SMU."""
54- logging .info ("Executing measurement: pin_name=%s voltage_level=%g" , pin_name , voltage_level )
54+ logging .info ("Executing measurement: pin_name=%s voltage_level=%g" , pin_names , voltage_level )
5555
5656 session_management_client = nims .session_management .Client (
5757 grpc_channel = measurement_service .get_channel (
@@ -64,7 +64,7 @@ def measure(
6464 reservation = stack .enter_context (
6565 session_management_client .reserve_sessions (
6666 context = measurement_service .context .pin_map_context ,
67- pin_names = [ pin_name ] ,
67+ pin_names = pin_names ,
6868 instrument_type_id = nims .session_management .INSTRUMENT_TYPE_NI_DCPOWER ,
6969 timeout = - 1 ,
7070 )
@@ -171,7 +171,10 @@ def _log_measured_values(measured_value, in_compliance):
171171
172172@click .command
173173@click .option (
174- "-v" , "--verbose" , count = True , help = "Enable verbose logging. Repeat to increase verbosity."
174+ "-v" ,
175+ "--verbose" ,
176+ count = True ,
177+ help = "Enable verbose logging. Repeat to increase verbosity." ,
175178)
176179@click .option (
177180 "--use-grpc-device/--no-use-grpc-device" ,
0 commit comments