Index: tests/dg_tests.py
===================================================================
diff -u -rdbad9b7e15a64dff87ea2be5c359de71b01ffa0f -r00fd6a0c7ad806801fc0d0c08f6e02b4591a6757
--- tests/dg_tests.py	(.../dg_tests.py)	(revision dbad9b7e15a64dff87ea2be5c359de71b01ffa0f)
+++ tests/dg_tests.py	(.../dg_tests.py)	(revision 00fd6a0c7ad806801fc0d0c08f6e02b4591a6757)
@@ -316,7 +316,10 @@
                     dg.dialysate_fill.total_volume))
     return info
 
+def get_hd_valves_info():
+    info = ('Air_trap' )
 
+
 def get_dg_idle_bad_fill_info():
     info = ('Bad_fill_state, {}, '
             .format(DGGenIdleModeBadFillSubStates(dg.gen_idle.sub_state).name))
@@ -494,6 +497,8 @@
 def run_heat_disinfect():
     complete_counter = 1
     f = open("/home/fw/projects/dialin/tests/Heat_disinfect.log", "w")
+    dg.heaters.cmd_heaters_broadcast_interval_override(50)
+    sleep(1)
     dg.hd_proxy.cmd_start_stop_dg_heat_disinfect()
 
     try:
@@ -515,21 +520,21 @@
 
             print(var)
             f.write(var)
-            sleep(1)
+            sleep(0.05)
 
             # If the mode came back to standby or standby solo
-            if dg.dg_operation_mode == 3 or dg.dg_operation_mode == 4:
+            #if dg.dg_operation_mode == 3 or dg.dg_operation_mode == 4:
                 # If it is the first call, stop heat disinfect
-                if complete_counter == 1:
-                    dg.hd_proxy.cmd_start_stop_dg_heat_disinfect(start=False)
+            #    if complete_counter == 1:
+            #        dg.hd_proxy.cmd_start_stop_dg_heat_disinfect(start=False)
 
                 # Write a few more complete states to make sure the complete state items are recorded
-                elif complete_counter == 3:
+            #    elif complete_counter == 3:
                     # pass
-                    f.close()
-                    break
+            #        f.close()
+            #        break
 
-                complete_counter += 1
+            #    complete_counter += 1
 
     except KeyboardInterrupt:
         dg.hd_proxy.cmd_start_stop_dg_heat_disinfect(start=False)
@@ -684,7 +689,7 @@
     hd.cmd_log_in_to_hd()
     sleep(1)
 
-    #run_heat_disinfect()
+    run_heat_disinfect()
 
     #run_flush_mode()
 
@@ -694,7 +699,7 @@
 
     # cmd_set_disinfect_ui_screen()
 
-    collect_treatment_data()
+    #collect_treatment_data()
 
     #collect_hd_treatment()