def test_get_dicom_field(self):
a = AnonymizeList(self.frame)
# Add a field
a.SetStringItems(['PatientName', 'PatientID'])
assert a.GetDicomField(0) == 'PatientName'
assert a.GetDicomField(1) == 'PatientID'
def test_set_replacement_dict_valid_fields(self):
a = AnonymizeList(self.frame)
d = {'Field': 'Value'}
# Ensure that the field to replace exists in the list
a.SetStringItems(d.keys())
a.SetReplacementDict(d)
# Ensure that the specified value was added to the second column
assert a.GetStringItem(0, 1) == 'Value'
def test_set_replacement_dict_invalid_field(self):
a = AnonymizeList(self.frame)
d = {'Invalid': 'value'}
# A control initialized with fields not in our dict
a.SetStringItems(['Field'])
a.SetReplacementDict(d)
assert a.GetStringItem(0, 1) == ''
assert a.ItemCount == 1
def test_patient_name_anonlist(self):
name = 'Patient0'
a = AnonymizeList(self.frame)
a.SetStringItems(['PatientName'])
a.SetReplacementDict({'PatientName': name})
dlg = QuickRenameDlg(self.frame, anonList=a)
expected = {
'PatientName': name,
'PatientID': '%(PatientName)s',
}
assert dlg.GetValues() == expected
def test_get_replacement_dict(self):
a = AnonymizeList(self.frame)
fields = {
'Field1': 'Replacement1',
'Field2': 'Replacement2',
'Field3': '',
}
a.SetStringItems(fields.keys())
a.SetReplacementDict(fields)
expected = {
'Field1': 'Replacement1',
'Field2': 'Replacement2',
}
d = a.GetReplacementDict()
assert d == expected
def test_no_patient_name_anonlist(self):
a = AnonymizeList(self.frame)
dlg = QuickRenameDlg(self.frame, anonList=a)
expected = {
'PatientName': '',
'PatientID': '%(PatientName)s',
}
assert dlg.GetValues() == expected
def test_on_accept(self):
name = 'Patient0'
a = AnonymizeList(self.frame)
a.SetStringItems(['PatientName', 'PatientID'])
a.SetReplacementDict({'PatientName': name, 'PatientID': 'ignored'})
dlg = QuickRenameDlg(self.frame, anonList=a)
dlg.OnAccept()
assert a.GetReplacementDict()['PatientID'] == '%(PatientName)s'
assert a.GetReplacementDict()['PatientName'] == name
def create(self):
vbox = wx.BoxSizer(wx.VERTICAL)
title = wx.StaticText(self, -1, "Fields to Omit")
vbox.Add(title, 0, wx.ALIGN_CENTER_HORIZONTAL | wx.TOP | wx.BOTTOM, 10)
self.anonList = AnonymizeList(self)
hbox = wx.BoxSizer(wx.HORIZONTAL)
self.store = wx.Button(self, -1, "Set as Default", size=(120, -1))
self.revert = wx.Button(self, -1, "Restore Default", size=(120, -1))
self.revert.Bind(wx.EVT_BUTTON, self.RevertState)
self.store.Bind(wx.EVT_BUTTON, self.SaveState)
hbox.Add(self.store, 0, wx.TOP | wx.LEFT, 10)
hbox.Add(self.revert, 0, wx.TOP | wx.LEFT, 10)
vbox.Add(self.anonList, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 10)
vbox.Add(hbox, 0, wx.ALIGN_CENTER_HORIZONTAL | wx.BOTTOM, 15)
self.SetSizer(vbox)
def test_get_anon_dict(self):
a = AnonymizeList(self.frame)
fields = {
'Field1': 'Replacement1',
'Field2': 'Replacement2',
'Field3': '',
}
keys = list(fields.keys())
keys.sort()
a.SetStringItems(keys)
a.SetReplacementDict(fields)
# Check fields 2 and 3
a.CheckItems([1, 2])
expected = {
'Field2': 'Replacement2',
'Field3': '',
}
assert a.GetAnonDict() == expected
def test_constructor(self):
a = AnonymizeList(self.frame)
assert isinstance(a, AnonymizeList)