def test_search_uid_list(self):
"""Test search using a UID list."""
query = Dataset()
query.QueryRetrieveLevel = 'PATIENT'
query.SOPInstanceUID = None
q = db._search_uid_list(query['SOPInstanceUID'], self.session)
assert 5 == len(q.all())
query.SOPInstanceUID = [
'1.3.6.1.4.1.5962.1.1.4.1.1.20040826185059.5457'
]
q = db._search_uid_list(query['SOPInstanceUID'], self.session)
assert 1 == len(q.all())
query.SOPInstanceUID = [
'1.3.6.1.4.1.5962.1.1.4.1.1.20040826185059.5457',
'1.3.6.1.4.1.5962.1.1.1.1.1.20040119072730.12322'
]
q = db._search_uid_list(query['SOPInstanceUID'], self.session)
assert 2 == len(q.all())
query.SOPInstanceUID = [
'1.3.6.1.4.1.5962.1.1.4.1.1.20040826185059.5457',
'1.3.6.1.4.1.5962.1.1.1.1.1.20040119072730.12322',
'1.3.46.423632.132218.1415242681.6'
]
q = db._search_uid_list(query['SOPInstanceUID'], self.session)
assert 2 == len(q.all())
def test_search_uid_list_empty(self):
"""Test searching an empty UID element works correctly."""
query = Dataset()
query.QueryRetrieveLevel = 'PATIENT'
query.SOPInstanceUID = None
q = db._search_uid_list(query['SOPInstanceUID'], self.session)
assert 5 == len(q.all())