Index: tests/test_dg_system_record.py
===================================================================
diff -u -r43e8345862c72340680c045e53b310aa291513a6 -r97c98d84604e3700dd446235e91a5ebe2db08022
--- tests/test_dg_system_record.py	(.../test_dg_system_record.py)	(revision 43e8345862c72340680c045e53b310aa291513a6)
+++ tests/test_dg_system_record.py	(.../test_dg_system_record.py)	(revision 97c98d84604e3700dd446235e91a5ebe2db08022)
@@ -10,21 +10,19 @@
     if dg.cmd_log_in_to_dg() == 0:
         exit(1)
     sleep(2)
+
     sys = dg.system_record
-    sys.DG_SYSTEM_RECORD_RECORD['system_record']['top_level_pn'][1] = 'DEN-SN-Y20'
-    print((sys.DG_SYSTEM_RECORD_RECORD))
-    #print(len(sys.DG_SYSTEM_RECORD_RECORD['system_record']['top_level_pn'][1]))
+    #sys.dg_system_record['system_record']['top_level_pn'][1] = 'DEN-SN-Y20'
+    print(sys.dg_system_record)
+    test = True
 
-    sys.get_dg_service_record()
+    if test:
+        sys.get_dg_system_record()
+        while True:
+            sleep(0.2)
+            if sys.current_message == sys.total_messages:
+                break
+    else:
+        sys.set_dg_system_record()
 
-    while True:
-        sleep(0.2)
-        if sys.current_message == sys.total_messages:
-            break
-
-    #print(sys.DG_SYSTEM_RECORD_RECORD)
-    print((sys.DG_SYSTEM_RECORD_RECORD['system_record']['top_level_pn'][1]))
-
-    #sys.set_dg_system_record()
-
-    #print(cal.DG_SYSTEM_RECORD_RECORD['system_record']['top_level_sn'][1])
+    print(sys.dg_system_record)