Index: tests/test_dg_records.py
===================================================================
diff -u -re8afa8d6323ede293e1fb094a461555fcf7ce0d9 -r8474f8e345f165187d4dde17840575ee4e98a9b4
--- tests/test_dg_records.py	(.../test_dg_records.py)	(revision e8afa8d6323ede293e1fb094a461555fcf7ce0d9)
+++ tests/test_dg_records.py	(.../test_dg_records.py)	(revision 8474f8e345f165187d4dde17840575ee4e98a9b4)
@@ -1,3 +1,18 @@
+###########################################################################
+#
+# Copyright (c) 2019-2021 Diality Inc. - All Rights Reserved.
+#
+# THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
+# WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
+#
+# @file    test_dg_records.py
+#
+# @author  (last)      Dara Navaei
+# @date    (last)      01-Jun-2021
+# @author  (original)  Dara Navaei
+# @date    (original)  01-Mar-2021
+#
+############################################################################
 from dialin.dg.dialysate_generator import DG
 from time import sleep
 
@@ -9,14 +24,15 @@
 
     try:
         if read:
-            cal.get_dg_calibration_record()
+            cal.cmd_request_dg_calibration_record()
             while True:
                 sleep(0.5)
                 if cal.cal_data != 0:
-                    if cal.is_reading_record_done():
-                        break
+                    #if cal.is_reading_record_done():
+                    sleep(4)
+                    break
         else:
-            cal.set_dg_calibration_record()
+            cal.cmd_set_dg_calibration_record()
 
         print(cal.dg_calibration_record)
 
@@ -76,12 +92,21 @@
         exit(1)
     sleep(2)
 
-    process_calibration_record(read=False)
+    dg.calibration_record.cmd_request_dg_calibration_record()
+    sleep(5)
+    dg.calibration_record.cmd_set_dg_calibration_record(dg.calibration_record.dg_calibration_record)
+    sleep(5)
+
+    dg.calibration_record.cmd_request_dg_calibration_record()
+    sleep(5)
+    dg.calibration_record.cmd_set_dg_calibration_record(dg.calibration_record.dg_calibration_record)
+    sleep(5)
+
+    #process_calibration_record(read=True)
     #process_system_record(read=True)
     #process_service_record(read=False)
 
 
 
 
 
-