Merge pull request #47 from Semi-ATE/issue_46

Fix for issue 41 : Missing SITE_NUM value in the PRR ATDF record
Semi-ATE · Jun 18, 2022 · fbde1fe · fbde1fe
2 parents 748bfe4 + 1652acf
commit fbde1fe
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/Semi_ATE/STDF/PRR.py b/Semi_ATE/STDF/PRR.py
@@ -84,7 +84,10 @@ def to_atdf(self):
 #       3 HEAD_NUM 
         body += self.gen_atdf(3)
 #       4 SITE_NUM 
-        body += self.gen_atdf(4)
+        site_num = self.gen_atdf(4)
+        if site_num == "|":
+            site_num = "1|"
+        body += site_num
 #       12 PART_ID 
         body += self.gen_atdf(12)
 #       6 NUM_TEST 

diff --git a/tests/test_PRR.py b/tests/test_PRR.py
@@ -29,7 +29,7 @@ def prr(endian):
     expected_atdf += str(head_num) +"|"
 
     site_num = 1
-    record.set_value('SITE_NUM', site_num)
+#    record.set_value('SITE_NUM', site_num)
     rec_len += 1;
     expected_atdf += str(site_num) + "|"
 
@@ -62,6 +62,8 @@ def prr(endian):
 
     '''
 
+#    part_flg = ['1', '0', '0', '1', '0', '0', '0', '0']
+#    part_flg = ['1', '0', '0', '0', '0', '0', '0', '0']
     part_flg = ['1', '0', '1', '1', '0', '0', '0', '0']
     record.set_value('PART_FLG', part_flg)
     rec_len += 1;
@@ -132,6 +134,13 @@ def prr(endian):
 
     w_data = record.__repr__()
     io_data = io.BytesIO(w_data)
+
+#    print("================")
+#    print(record)
+#    print("================")
+#    print(record.to_atdf())
+#    print("================")
+
 
     stdfRecTest = STDFRecordTest(io_data, endian)
 #   rec_len, rec_type, rec_sub