@sh.add_function(dsp, outputs=['drive_battery_loads'])
def calculate_drive_battery_loads_v1(times, drive_battery_load):
"""
Calculates drive battery load vector [kW].
:param times:
Time vector [s].
:type times: numpy.array
:param drive_battery_load:
Drive electric load [kW].
:type drive_battery_load: float
:return:
Drive battery load vector [kW].
:rtype: numpy.array
"""
return np.tile(drive_battery_load, times.shape)