Index: tests/dg_tests.py
===================================================================
diff -u -r1ceb44f1f6d97ce295db7c3233043df9cca35746 -raa38884ffae56e67897e1ea53108c4a950bdd998
--- tests/dg_tests.py	(.../dg_tests.py)	(revision 1ceb44f1f6d97ce295db7c3233043df9cca35746)
+++ tests/dg_tests.py	(.../dg_tests.py)	(revision aa38884ffae56e67897e1ea53108c4a950bdd998)
@@ -29,7 +29,7 @@
 from dialin.hd.temperatures import HDTemperaturesNames
 from dialin.dg.concentrate_pumps import DGConcentratePumpsStates
 from dialin.dg.uv_reactors import ReactorsNames
-from dialin.common.hd_defs import HDOpModes, PreTreatmentWetSelfTestStates, PostTreatmentStates
+from dialin.common.hd_defs import HDOpModes, HDStandbyStates, PreTreatmentWetSelfTestStates, PostTreatmentStates
 from dialin.hd.post_treatment import HDPostTreatmentDrainStates
 from dialin.common.dg_defs import DGEventList
 from dialin.common.hd_defs import HDEventList
@@ -515,9 +515,8 @@
             dg_fans = get_dg_fans_info()
             hd_fans = get_hd_fans_info()
             conc = get_concentrate_pumps_info()
-            uv = get_uv_reactors_info()
 
-            var = disinfect + load_cell + drain + ro + temp + heaters + uv + dg_fans + hd_fans + valves + conc + uv + '\r'
+            var = disinfect + load_cell + drain + ro + temp + heaters + uv + dg_fans + hd_fans + valves + conc + '\r'
 
             print(var)
             f.write(var)
@@ -690,7 +689,7 @@
     hd.cmd_log_in_to_hd()
     sleep(1)
 
-    run_heat_disinfect()
+    #run_heat_disinfect()
 
     #run_flush_mode()
 
@@ -700,7 +699,7 @@
 
     # cmd_set_disinfect_ui_screen()
 
-    #collect_treatment_data()
+    collect_treatment_data()
 
     #collect_hd_treatment()
 
@@ -740,3 +739,5 @@
     #    print(get_ro_info())
     #    sleep(1)
 
+    #dg.heaters.cmd_set_primary_and_trimmer_heaters_target_temperature(primary_target_temp=40, trimmer_target_temp=52)
+