def test_patients_factory(self):
patients = patients_factory(self.gen.root)
patient_1 = patients[0]
patient_2 = patients[1]
self.assertEqual(len(patients), 2)
self.assertEqual(patient_1.id, '0')
self.assertEqual(patient_2.id, '1')
self.assertEqual(patient_1.patient_id, 'nhc_demo_patient_0')
self.assertEqual(patient_2.patient_id, 'nhc_demo_patient_1')
self.assertEqual(patient_1.placement_id, 'nhc_demo_placement_0')
self.assertEqual(patient_2.placement_id, 'nhc_demo_placement_1')
self.assertEqual(patient_1.activity_id,
'nhc_activity_demo_placement_0')
self.assertEqual(patient_2.activity_id,
'nhc_activity_demo_placement_1')
self.assertEqual(patient_1.spell_activity_id,
'nhc_activity_demo_spell_0')
self.assertEqual(patient_2.spell_activity_id,
'nhc_activity_demo_spell_1')
def setUp(self):
"""
WardStrategy needs patient placements.
"""
bed_patient = PatientsGenerator(0, 2, 0, 'a')
self.gen = PlacementsGenerator(bed_patient, [-1, -2])
self.patients = patients_factory(self.gen.root)
risk = {'high': 1, 'medium': 2, 'low': 10, 'none': 15}
self.user_ids = ['user_1', 'user_2']
self.strategy = WardStrategy(self.patients, self.user_ids, risk, 1, 1,
[30])
def __init__(self, wards, beds_per_ward, bed_patient_per_ward,
non_bed_patient_per_ward, users_schema, data_folder):
total_patients_per_ward = \
bed_patient_per_ward + non_bed_patient_per_ward
patient_id_offset = 1
# wards = ['a', 'b', 'c', 'd', 'e']
# bed_patients_per_ward = 28
# non_bed_patients_per_ward = 12
# List of time periods to randomly offset admissions
self.admit_offset_list = ['-1', '-2']
self.admit_date_eval_string = '(datetime.now() + timedelta({0}))' \
'.strftime(\'%Y-%m-%d %H:%M:%S\')'
# Regex to use to get the ID for a patient from id attribute on record
patient_id_regex_string = r'nhc_demo_patient_(\d+)'
self.patient_id_regex = re.compile(patient_id_regex_string)
point_of_service = POSGenerator()
self.indent(point_of_service.root)
users_tree = ElementTree(point_of_service.root)
users_tree.write(os.path.join(data_folder, 'pos.xml'))
users_generator = UsersGenerator(users_schema)
users_generator.generate_adt_user()
users_generator.generate_multi_wards_users(wards)
users_generator.generate_users_not_assigned()
self.indent(users_generator.class_root)
users_tree = ElementTree(users_generator.class_root)
users_tree.write(os.path.join(data_folder, 'users.xml'))
# Generate demo data for each ward,
# with files named after different type of data,
# grouped in one folder for each ward
self.ews_seq = 0
self.assess_seq = 0
self.medical_seq = 0
for index, ward in enumerate(wards):
# Locations demo data
locations = LocationsGenerator(ward, beds_per_ward)
# Users demo data
users_per_ward_root = users_generator.generate_users_per_ward(
ward, beds_per_ward)
# Patients demo data
patients = PatientsGenerator(
(index * total_patients_per_ward) + patient_id_offset,
bed_patient_per_ward,
non_bed_patient_per_ward,
ward
)
offsets = [random.choice(
self.admit_offset_list) for _ in range(
len(patients.data.findall('record')))]
# Spells demo data
spells = SpellsGenerator(patients, offsets)
# Admissions demo data
admissions = AdmissionsGenerator(patients, offsets)
# Placements demo data
placements = PlacementsGenerator(patients, offsets)
# Strategy
risk_distribution = {
'high': 0, 'medium': 2, 'low': 11, 'none': 15
}
# 50% observations are overdue by default
overdue_ratio = 0.5
# all overdue observations are within 30 mins overdue
overdue_distribution = [30]
if index == 0:
# ICU
risk_distribution = {
'high': 3, 'medium': 4, 'low': 20, 'none': 1
}
elif index == 1:
# 75% observations are overdue, 1/3 overdue by 60 mins, the
# rest overdue by, at most, 30 mins
overdue_ratio = 0.75
overdue_distribution = [30, 30, 60]
risk_distribution = {
'high': 0, 'medium': 0, 'low': 20, 'none': 8
}
elif index == 4:
# all observations are on-time
overdue_ratio = 0
risk_distribution = {
'high': 0, 'medium': 0, 'low': 0, 'none': 28
}
news_patients = patients_factory(placements.root)
hca_nurse_ids = get_hca_nurse_users(users_per_ward_root)
partial_news_per_patient = 1
# create ward strategy here
ward_strategy = WardStrategy(
news_patients, hca_nurse_ids, risk_distribution,
partial_news_per_patient, overdue_ratio, overdue_distribution
)
# NEWS demo data
news = NewsGenerator(ward_strategy, self.ews_seq, self.assess_seq,
self.medical_seq)
self.ews_seq = news.ews_seq
self.assess_seq = news.assess_seq
self.medical_seq = news.medical_seq
# Pretty format the XML trees
self.indent(locations.root)
self.indent(users_per_ward_root)
self.indent(patients.root)
self.indent(spells.root)
self.indent(admissions.root)
self.indent(placements.root)
self.indent(news.root)
# Actually write the XML files (creating them if needed)
ward_folder = os.path.join(data_folder, 'ward_{0}'.format(ward))
if not os.path.isdir(ward_folder):
os.mkdir(ward_folder)
locations_tree = ElementTree(locations.root)
locations_tree.write(os.path.join(ward_folder,
'demo_locations.xml'))
users_per_ward_tree = ElementTree(users_per_ward_root)
users_per_ward_tree.write(os.path.join(ward_folder,
'demo_users.xml'))
patients_tree = ElementTree(patients.root)
patients_tree.write(os.path.join(ward_folder, 'demo_patients.xml'))
spells_tree = ElementTree(spells.root)
spells_tree.write(os.path.join(ward_folder, 'demo_spells.xml'))
admissions_tree = ElementTree(admissions.root)
admissions_tree.write(os.path.join(ward_folder,
'demo_admissions.xml'))
placements_tree = ElementTree(placements.root)
placements_tree.write(os.path.join(ward_folder,
'demo_placements.xml'))
news_tree = ElementTree(news.root)
news_tree.write(os.path.join(ward_folder, 'demo_news.xml'))