def test_add_tool_elements(self):
device = Device(name="TestDevice")
extruder = Tool(name="extruder1", category="extruder")
stepper = Actuator(name="extruder_1_stepper", category="steppers")
heater = Actuator(name="extruder_1_heater", category="heater")
fan = Actuator(name="extruder_1_fan", category="fan")
temp_sensor = Sensor(name="extruder_1_temperature_sensor", category="temperature")
extruder.add_children([stepper, heater, fan, temp_sensor])
device.add_child(extruder)
self.assertEquals(device.children_components, [extruder])
self.assertEquals(extruder.children_components, [stepper, heater, fan, temp_sensor])
def test_get_components_by_category_recursive(self):
device = Device(name="TestDevice")
extruder = Tool(name="extruder1", category="extruder")
stepper = Actuator(name="extruder_1_stepper", category="steppers")
stepper2 = Actuator(name="extruder_1_stepper2", category="steppers")
heater = Actuator(name="extruder_1_heater", category="heater")
fan = Actuator(name="extruder_1_fan", category="fan")
temp_sensor = Sensor(name="extruder_1_temp_sensor", category="temperature")
extruder.add_children([stepper, stepper2, heater, fan, temp_sensor])
device.add_child(extruder)
obs_components = device.get_children_bycategory(category="steppers", recursive=True)
exp_components = [stepper, stepper2]
self.assertEquals(obs_components, exp_components)
def test_add_tool(self):
device = Device(name="TestDevice")
extruder = Tool(name="extruder1", category="extruder")
device.add_child(extruder)
self.assertEquals(device.children_components, [extruder])
