def test_code_dataelem_sequence(self):
"""Test utils.codify.code_dataelem"""
# ControlPointSequence
elem = DataElement(0x300A0111, 'SQ', [])
out = "\n# Control Point Sequence\n"
out += "cp_sequence = Sequence()\n"
out += "ds.ControlPointSequence = cp_sequence"
assert out == code_dataelem(elem)
def test_code_dataelem_sequence(self):
"""Test utils.codify.code_dataelem"""
# ControlPointSequence
elem = DataElement(0x300A0111, 'SQ', [])
out = "\n# Control Point Sequence\n"
out += "cp_sequence = Sequence()\n"
out += "ds.ControlPointSequence = cp_sequence"
assert code_dataelem(elem) == out
def test_code_dataelem_standard(self):
"""Test utils.codify.code_dataelem for standard element"""
# Element keyword in data dictionary
input_elem = [DataElement(0x00100010, 'PN', 'CITIZEN'),
DataElement(0x0008010c, 'UI', '1.1.2.3.4.5'),
DataElement(0x00080301, 'US', 1200)]
out_str = ["ds.PatientName = 'CITIZEN'",
"ds.CodingSchemeUID = '1.1.2.3.4.5'",
"ds.PrivateGroupReference = 1200"]
for elem, out in zip(input_elem, out_str):
assert out == code_dataelem(elem)
def test_code_dataelem_private(self):
"""Test utils.codify.code_dataelem"""
# Element keyword not in data dictionary
input_elem = [DataElement(0x00111010, 'PN', 'CITIZEN'),
DataElement(0x0081010c, 'UI', '1.1.2.3.4.5'),
DataElement(0x11110301, 'US', 1200)]
out_str = ["ds.add_new((0x0011, 0x1010), 'PN', 'CITIZEN')",
"ds.add_new((0x0081, 0x010c), 'UI', '1.1.2.3.4.5')",
"ds.add_new((0x1111, 0x0301), 'US', 1200)"]
for elem, out in zip(input_elem, out_str):
assert out == code_dataelem(elem)
def test_code_dataelem_private(self):
"""Test utils.codify.code_dataelem"""
# Element keyword not in data dictionary
input_elem = [DataElement(0x00111010, 'PN', 'CITIZEN'),
DataElement(0x0081010c, 'UI', '1.1.2.3.4.5'),
DataElement(0x11110301, 'US', 1200)]
out_str = ["ds.add_new((0x0011, 0x1010), 'PN', 'CITIZEN')",
"ds.add_new((0x0081, 0x010c), 'UI', '1.1.2.3.4.5')",
"ds.add_new((0x1111, 0x0301), 'US', 1200)"]
for elem, out in zip(input_elem, out_str):
assert code_dataelem(elem) == out
def test_code_dataelem_standard(self):
"""Test utils.codify.code_dataelem for standard element"""
# Element keyword in data dictionary
input_elem = [DataElement(0x00100010, 'PN', 'CITIZEN'),
DataElement(0x0008010c, 'UI', '1.1.2.3.4.5'),
DataElement(0x00080301, 'US', 1200)]
out_str = ["ds.PatientName = 'CITIZEN'",
"ds.CodingSchemeUID = '1.1.2.3.4.5'",
"ds.PrivateGroupReference = 1200"]
for elem, out in zip(input_elem, out_str):
assert code_dataelem(elem) == out
def test_code_dataelem_exclude_size(self):
"""Test utils.codify.code_dataelem exclude_size param"""
input_elem = [DataElement(0x00100010, 'OB', 'CITIZEN'),
DataElement(0x0008010c, 'UI', '1.1')]
# Fails
# DataElement(0x00080301, 'US', 1200)]
out_str = ["ds.PatientName = # XXX Array of 7 bytes excluded",
"ds.CodingSchemeUID = '1.1'"]
# Fails
# "ds.PrivateGroupReference = 1200"]
for elem, out in zip(input_elem, out_str):
assert code_dataelem(elem, exclude_size=4) == out
def test_code_sequence(self):
"""Test utils.codify.code_dataelem"""
# ControlPointSequence
elem = DataElement(0x300A0111, 'SQ', [])
elem.value.append(Dataset())
elem[0].PatientID = '1234'
out = "\n"
out += "# Control Point Sequence\n"
out += "cp_sequence = Sequence()\n"
out += "ds.ControlPointSequence = cp_sequence\n"
out += "\n"
out += "# Control Point Sequence: Control Point 1\n"
out += "cp1 = Dataset()\n"
out += "cp1.PatientID = '1234'\n"
out += "cp_sequence.append(cp1)"
assert code_dataelem(elem) == out
def test_code_sequence(self):
"""Test utils.codify.code_dataelem"""
# ControlPointSequence
elem = DataElement(0x300A0111, 'SQ', [])
elem.value.append(Dataset())
elem[0].PatientID = '1234'
out = "\n"
out += "# Control Point Sequence\n"
out += "cp_sequence = Sequence()\n"
out += "ds.ControlPointSequence = cp_sequence\n"
out += "\n"
out += "# Control Point Sequence: Control Point 1\n"
out += "cp1 = Dataset()\n"
out += "cp1.PatientID = '1234'\n"
out += "cp_sequence.append(cp1)"
assert code_dataelem(elem) == out
