@sh.add_function(dsp, outputs=['drive_battery_voltages'])
def calculate_drive_battery_voltages_v1(
drive_battery_electric_powers, drive_battery_model):
"""
Calculate the drive battery current vector [A].
:param drive_battery_electric_powers:
Drive battery electric power [kW].
:type drive_battery_electric_powers: numpy.array
:param drive_battery_model:
Drive battery current model.
:type drive_battery_model: DriveBatteryModel
:return:
Drive battery current vector [A].
:rtype: numpy.array
"""
return drive_battery_model.voltages(drive_battery_electric_powers)