Index: tests/test_gen_requirements.py =================================================================== diff -u -r8ea13ae6dd10732bfcc456798f4785c4d88c95d3 -rb9fb7db20f9ac97da147afc08a18648347dc8d90 --- tests/test_gen_requirements.py (.../test_gen_requirements.py) (revision 8ea13ae6dd10732bfcc456798f4785c4d88c95d3) +++ tests/test_gen_requirements.py (.../test_gen_requirements.py) (revision b9fb7db20f9ac97da147afc08a18648347dc8d90) @@ -83,16 +83,15 @@ return result -def generate_requirements(): + +def generate_hd_requirements(): """ - Generates requires for both HD and DG + Generates requires for the HD @return: None """ hd = HD() - result = {} - modules = [ hd.alarms, hd.buttons, @@ -109,10 +108,16 @@ gen_publish_requirements(modules) gen_cmd_requirements(modules) - dg = DG() - result = {} +def generate_dg_requirements(): + """ + Generates requires for the DG + @return: None + """ + + dg = DG() + modules = [ dg.hd_proxy, dg.load_cells, @@ -121,13 +126,15 @@ dg.valves, dg.ro_pump, dg.drain_pump, - # dg.heaters, - # dg.temperature_sensors + dg.heaters, + dg.temperature_sensors, + dg.conductivity_sensors ] gen_publish_requirements(modules) gen_cmd_requirements(modules) if __name__ == '__main__': - generate_requirements() + generate_dg_requirements() + generate_hd_requirements()