def test_get_private_diagnostic_test_cases_from_episode(self):
self.create_case_structure()
test1 = self.create_test_case(
self.occurrence_id, {
'enrolled_in_private': 'true',
'date_reported': '2017-08-14',
'purpose_of_test': 'diagnostic',
'investigation_id': 'ABC-ABC-ABC',
'result_grade': 'TB Detected: 3+ scanty'
})
test2 = self.create_test_case(
self.occurrence_id, {
'enrolled_in_private': 'true',
'date_reported': '2017-08-15',
'purpose_of_test': 'diagnostic',
'investigation_id': 'DEF-DEF-DEF',
'result_grade': 'TB Detected: 3+ scanty',
})
self.create_test_case(
self.occurrence_id, {
'enrolled_in_private': 'false',
'date_reported': '2017-08-15',
'purpose_of_test': 'diagnostic',
'investigation_id': 'DEF-DEF-DEF',
'result_grade': 'TB Detected: 3+ scanty',
})
self.assertItemsEqual([test1, test2],
get_private_diagnostic_test_cases_from_episode(
self.domain, self.episode_id))
def diagnostic_tests(self):
try:
return get_private_diagnostic_test_cases_from_episode(self.domain, self.episode.case_id)
except ENikshayCaseNotFound:
return None
