def test_AnalogInputNode_units():
node = AnalogInputNode()
node.units = 'mTorr'
assert node.units == 'mTorr'
node.units = 'Torr'
assert node.units == 'Torr'
def test_AnalogInputNode_displayScientific():
node = AnalogInputNode()
assert node.displayScientific == False
node.displayScientific = True
assert node.displayScientific == True
node.displayScientific = 'ham'
assert node.displayScientific == False
def test_AnalogInputNode_displayDigits():
node = AnalogInputNode()
assert node.displayDigits == strings.A_DISPLAY_DIGITS_DEFAULT
node.displayDigits = -1
assert node.displayDigits == 0
node.displayDigits = 9991
assert node.displayDigits == strings.A_DISPLAY_DIGITS_MAX
def test_AnalogInputNode_displayToHal():
node = AnalogInputNode()
cal = [['hal_value', 'gui_value'], [0.0, 0.0], [10.0, 500.0]]
node.calibrationTableData = cal
node.setData(col.HAL_VALUE, 1.75)
assert node.data(col.HAL_VALUE) == 1.75
assert node.displayToHal(node.value()) == 1.75
def test_AnalogInputNode_calibrationTableData():
node = AnalogInputNode()
good_cal = [['hal_value', 'gui_value'], [0.0, 0.0], [1.0, 12.4], [2, 24.8]]
bad_cal = [['hal_value', 'gui_value'], [0.0, 0.0], [-1.0, 12.4], [2, 24.8]]
node.calibrationTableData = good_cal
assert node.calibrationTableData == good_cal
node.calibrationTableData = bad_cal
assert node.calibrationTableData == good_cal
node.setData(col.HAL_VALUE, .12)
assert node.value() == 1.488
assert node.data(col.VALUE) == 1.488
def insertChild(self, parent_index, child_type, insert_row=None):
parent_node = parent_index.internalPointer()
#allowed_rows = self.allowedInsertRows(parent_index, child_type)
if insert_row is None:
insert_row = parent_index.internalPointer().childCount()
#if not allowed_rows:
# msg = "Child of type " + child_type + " cannot be inserted into " + parent_node.typeInfo()
# MessageBox(msg)
# return False
#if preferred_row is None:
# insert_row = allowed_rows[-1]
#elif preferred_row in allowed_rows:
# insert_row = preferred_row
#else:
# MessageBox("Attempting to insert child in tool model at bad row", "row: ", preferred_row, "allowed rows: ", allowed_rows)
# return False
if insert_row is not False:
self.beginInsertRows(parent_index, insert_row, insert_row)
if child_type == typ.TOOL_NODE:
parent_node.insertChild(insert_row, ToolNode())
elif child_type == typ.SYSTEM_NODE:
parent_node.insertChild(insert_row, SystemNode())
elif child_type == typ.DEVICE_NODE:
parent_node.insertChild(insert_row, DeviceNode())
elif child_type == typ.DEVICE_ICON_NODE:
parent_node.insertChild(insert_row, DeviceIconNode())
elif child_type == typ.D_IN_NODE:
parent_node.insertChild(insert_row, DigitalInputNode())
elif child_type == typ.D_OUT_NODE:
parent_node.insertChild(insert_row, DigitalOutputNode())
elif child_type == typ.A_IN_NODE:
parent_node.insertChild(insert_row, AnalogInputNode())
elif child_type == typ.A_OUT_NODE:
parent_node.insertChild(insert_row, AnalogOutputNode())
elif child_type == typ.BOOL_VAR_NODE:
parent_node.insertChild(insert_row, BoolVarNode())
elif child_type == typ.FLOAT_VAR_NODE:
parent_node.insertChild(insert_row, FloatVarNode())
else:
MessageBox('Attempting to insert unknown node of type',
child_type)
'''
Add code here to add said child to the behavior tree
'''
self.endInsertRows()
new_child_index = self.index(insert_row, 0, parent_index)
#Add the node to the behavior tree
try:
device_bt_model = new_child_index.internalPointer().parent(
).behaviorTreeModel()
device_bt_model.addNode(new_child_index.internalPointer())
except:
pass
#for set_index in device_bt_model.indexesOfType(typ.SET_NODE):
# print("hmmm - ", new_child_index)
# set_index.internalPointer().addNode(new_child_index.internalPointer())
return new_child_index
def test_AnalogInputNode_setData():
node = AnalogInputNode()
cal = [['hal_value', 'gui_value'], [0.0, 0.0], [10.0, 500.0]]
node.calibrationTableData = cal
node.setData(col.HAL_VALUE, 1.5)
node.setData(col.VALUE, "can't set this")
node.setData(col.UNITS, 'mTorr')
node.setData(col.DISPLAY_DIGITS, 5)
node.setData(col.DISPLAY_SCIENTIFIC, True)
node.setData(col.CALIBRATION_TABLE_MODEL, "can't set this")
assert node.data(col.HAL_VALUE) == 1.5
assert node.data(col.VALUE) == 75
assert node.data(col.UNITS) == 'mTorr'
assert node.data(col.DISPLAY_DIGITS) == 5
assert node.data(col.DISPLAY_SCIENTIFIC) == True
assert type(node.data(col.CALIBRATION_TABLE_MODEL)) == type(
CalibrationTableModel())
def test_AnalogInputNode_value():
node = AnalogInputNode()
cal = [['hal_value', 'gui_value'], [0.0, 0.0], [10.0, 500.0]]
node.calibrationTableData = cal
node.setData(col.HAL_VALUE, 1.0)
assert node.value() == 50
assert node.data(col.VALUE) == 50
node.setData(col.HAL_VALUE, 1.5)
assert node.value() == 75
assert node.data(col.VALUE) == 75
def test_AnalogInputNode_calibrationTableModel():
node = AnalogInputNode()
assert type(node.calibrationTableModel()) == type(CalibrationTableModel())
def test_AnalogInputNode_typeInfo():
node = AnalogInputNode()
assert node.typeInfo() == typ.A_IN_NODE