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])