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