-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path_Github_synthpops_synthpops_long_term_care_facilities_py.html
617 lines (617 loc) · 185 KB
/
_Github_synthpops_synthpops_long_term_care_facilities_py.html
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
<!DOCTYPE html>
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
<meta http-equiv="X-UA-Compatible" content="IE=emulateIE7" />
<title>Coverage for D:\Github\synthpops\synthpops\long_term_care_facilities.py: 84%</title>
<link rel="stylesheet" href="style.css" type="text/css">
<script type="text/javascript" src="jquery.min.js"></script>
<script type="text/javascript" src="jquery.hotkeys.js"></script>
<script type="text/javascript" src="jquery.isonscreen.js"></script>
<script type="text/javascript" src="coverage_html.js"></script>
<script type="text/javascript">
jQuery(document).ready(coverage.pyfile_ready);
</script>
</head>
<body class="pyfile">
<div id="header">
<div class="content">
<h1>Coverage for <b>D:\Github\synthpops\synthpops\long_term_care_facilities.py</b> :
<span class="pc_cov">84%</span>
</h1>
<img id="keyboard_icon" src="keybd_closed.png" alt="Show keyboard shortcuts" />
<h2 class="stats">
322 statements
<button type="button" class="run shortkey_r button_toggle_run" title="Toggle lines run">271 run</button>
<button type="button" class="mis show_mis shortkey_m button_toggle_mis" title="Toggle lines missing">51 missing</button>
<button type="button" class="exc show_exc shortkey_x button_toggle_exc" title="Toggle lines excluded">0 excluded</button>
</h2>
</div>
</div>
<div class="help_panel">
<img id="panel_icon" src="keybd_open.png" alt="Hide keyboard shortcuts" />
<p class="legend">Hot-keys on this page</p>
<div>
<p class="keyhelp">
<span class="key">r</span>
<span class="key">m</span>
<span class="key">x</span>
<span class="key">p</span> toggle line displays
</p>
<p class="keyhelp">
<span class="key">j</span>
<span class="key">k</span> next/prev highlighted chunk
</p>
<p class="keyhelp">
<span class="key">0</span> (zero) top of page
</p>
<p class="keyhelp">
<span class="key">1</span> (one) first highlighted chunk
</p>
</div>
</div>
<div id="source">
<p id="t1" class="pln"><span class="n"><a href="#t1">1</a></span><span class="t"><span class="str">"""</span> </span><span class="r"></span></p>
<p id="t2" class="pln"><span class="n"><a href="#t2">2</a></span><span class="t"><span class="str">Modeling Seattle Metro Long Term Care Facilities</span> </span><span class="r"></span></p>
<p id="t3" class="pln"><span class="n"><a href="#t3">3</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t4" class="pln"><span class="n"><a href="#t4">4</a></span><span class="t"><span class="str">"""</span> </span><span class="r"></span></p>
<p id="t5" class="pln"><span class="n"><a href="#t5">5</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t6" class="run"><span class="n"><a href="#t6">6</a></span><span class="t"><span class="key">import</span> <span class="nam">numpy</span> <span class="key">as</span> <span class="nam">np</span> </span><span class="r"></span></p>
<p id="t7" class="pln"><span class="n"><a href="#t7">7</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t8" class="run"><span class="n"><a href="#t8">8</a></span><span class="t"><span class="key">import</span> <span class="nam">os</span> </span><span class="r"></span></p>
<p id="t9" class="run"><span class="n"><a href="#t9">9</a></span><span class="t"><span class="key">import</span> <span class="nam">math</span> </span><span class="r"></span></p>
<p id="t10" class="run"><span class="n"><a href="#t10">10</a></span><span class="t"><span class="key">from</span> <span class="nam">copy</span> <span class="key">import</span> <span class="nam">deepcopy</span> </span><span class="r"></span></p>
<p id="t11" class="run"><span class="n"><a href="#t11">11</a></span><span class="t"><span class="key">import</span> <span class="nam">matplotlib</span><span class="op">.</span><span class="nam">pyplot</span> <span class="key">as</span> <span class="nam">plt</span> </span><span class="r"></span></p>
<p id="t12" class="run"><span class="n"><a href="#t12">12</a></span><span class="t"><span class="key">from</span> <span class="nam">collections</span> <span class="key">import</span> <span class="nam">Counter</span> </span><span class="r"></span></p>
<p id="t13" class="pln"><span class="n"><a href="#t13">13</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t14" class="run"><span class="n"><a href="#t14">14</a></span><span class="t"><span class="key">from</span> <span class="op">.</span> <span class="key">import</span> <span class="nam">base</span> <span class="key">as</span> <span class="nam">spb</span> </span><span class="r"></span></p>
<p id="t15" class="run"><span class="n"><a href="#t15">15</a></span><span class="t"><span class="key">from</span> <span class="op">.</span> <span class="key">import</span> <span class="nam">data_distributions</span> <span class="key">as</span> <span class="nam">spdata</span> </span><span class="r"></span></p>
<p id="t16" class="run"><span class="n"><a href="#t16">16</a></span><span class="t"><span class="key">from</span> <span class="op">.</span> <span class="key">import</span> <span class="nam">sampling</span> <span class="key">as</span> <span class="nam">spsamp</span> </span><span class="r"></span></p>
<p id="t17" class="run"><span class="n"><a href="#t17">17</a></span><span class="t"><span class="key">from</span> <span class="op">.</span> <span class="key">import</span> <span class="nam">contacts</span> <span class="key">as</span> <span class="nam">spct</span> </span><span class="r"></span></p>
<p id="t18" class="run"><span class="n"><a href="#t18">18</a></span><span class="t"><span class="key">from</span> <span class="op">.</span> <span class="key">import</span> <span class="nam">contact_networks</span> <span class="key">as</span> <span class="nam">spcnx</span> </span><span class="r"></span></p>
<p id="t19" class="run"><span class="n"><a href="#t19">19</a></span><span class="t"><span class="key">from</span> <span class="op">.</span> <span class="key">import</span> <span class="nam">read_write</span> <span class="key">as</span> <span class="nam">sprw</span> </span><span class="r"></span></p>
<p id="t20" class="pln"><span class="n"><a href="#t20">20</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t21" class="pln"><span class="n"><a href="#t21">21</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t22" class="pln"><span class="n"><a href="#t22">22</a></span><span class="t"><span class="com"># Customized age resampling method</span> </span><span class="r"></span></p>
<p id="t23" class="run"><span class="n"><a href="#t23">23</a></span><span class="t"><span class="key">def</span> <span class="nam">custom_resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t24" class="pln"><span class="n"><a href="#t24">24</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
<p id="t25" class="pln"><span class="n"><a href="#t25">25</a></span><span class="t"><span class="str"> Resampling younger ages to better match data</span> </span><span class="r"></span></p>
<p id="t26" class="pln"><span class="n"><a href="#t26">26</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t27" class="pln"><span class="n"><a href="#t27">27</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
<p id="t28" class="pln"><span class="n"><a href="#t28">28</a></span><span class="t"><span class="str"> single_year_age_distr (dict) : age distribution</span> </span><span class="r"></span></p>
<p id="t29" class="pln"><span class="n"><a href="#t29">29</a></span><span class="t"><span class="str"> age (int) : age as an integer</span> </span><span class="r"></span></p>
<p id="t30" class="pln"><span class="n"><a href="#t30">30</a></span><span class="t"><span class="str"> Returns:</span> </span><span class="r"></span></p>
<p id="t31" class="pln"><span class="n"><a href="#t31">31</a></span><span class="t"><span class="str"> Resampled age as an integer.</span> </span><span class="r"></span></p>
<p id="t32" class="pln"><span class="n"><a href="#t32">32</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t33" class="pln"><span class="n"><a href="#t33">33</a></span><span class="t"><span class="str"> Notes:</span> </span><span class="r"></span></p>
<p id="t34" class="pln"><span class="n"><a href="#t34">34</a></span><span class="t"><span class="str"> This is not always necessary, but is mostly used to smooth out sharp edges in the age distribution when spsamp.resample_age() produces too many of one year and under produces the surrounding ages. For example, new borns (0 years old) may</span> </span><span class="r"></span></p>
<p id="t35" class="pln"><span class="n"><a href="#t35">35</a></span><span class="t"><span class="str"> be over produced, and 1 year olds under produced, so this function can be customized to correct for that.</span> </span><span class="r"></span></p>
<p id="t36" class="pln"><span class="n"><a href="#t36">36</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
<p id="t37" class="pln"><span class="n"><a href="#t37">37</a></span><span class="t"> <span class="com"># exp_age_distr = np.array(list(exp_age_distr_dict.values()), dtype=np.float64)</span> </span><span class="r"></span></p>
<p id="t38" class="run"><span class="n"><a href="#t38">38</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t39" class="run"><span class="n"><a href="#t39">39</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op">==</span> <span class="num">7</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t40" class="run"><span class="n"><a href="#t40">40</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="num">0.25</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t41" class="run"><span class="n"><a href="#t41">41</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t42" class="run"><span class="n"><a href="#t42">42</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op">==</span> <span class="num">6</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t43" class="run"><span class="n"><a href="#t43">43</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="num">0.25</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t44" class="run"><span class="n"><a href="#t44">44</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t45" class="run"><span class="n"><a href="#t45">45</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op">==</span> <span class="num">5</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t46" class="run"><span class="n"><a href="#t46">46</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="num">0.2</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t47" class="run"><span class="n"><a href="#t47">47</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t48" class="run"><span class="n"><a href="#t48">48</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op">==</span> <span class="num">0</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t49" class="run"><span class="n"><a href="#t49">49</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="num">0.0</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t50" class="mis show_mis"><span class="n"><a href="#t50">50</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t51" class="run"><span class="n"><a href="#t51">51</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op">==</span> <span class="num">1</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t52" class="run"><span class="n"><a href="#t52">52</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="num">0.1</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t53" class="run"><span class="n"><a href="#t53">53</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t54" class="run"><span class="n"><a href="#t54">54</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op">==</span> <span class="num">2</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t55" class="run"><span class="n"><a href="#t55">55</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="num">0.0</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t56" class="mis show_mis"><span class="n"><a href="#t56">56</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t57" class="run"><span class="n"><a href="#t57">57</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op">==</span> <span class="num">4</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t58" class="run"><span class="n"><a href="#t58">58</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="num">0.1</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t59" class="run"><span class="n"><a href="#t59">59</a></span><span class="t"> <span class="nam">a</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">resample_age</span><span class="op">(</span><span class="nam">exp_age_distr</span><span class="op">,</span> <span class="nam">a</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t60" class="run"><span class="n"><a href="#t60">60</a></span><span class="t"> <span class="key">return</span> <span class="nam">a</span> </span><span class="r"></span></p>
<p id="t61" class="pln"><span class="n"><a href="#t61">61</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t62" class="pln"><span class="n"><a href="#t62">62</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t63" class="pln"><span class="n"><a href="#t63">63</a></span><span class="t"><span class="com"># Customized household construction methods</span> </span><span class="r"></span></p>
<p id="t64" class="run"><span class="n"><a href="#t64">64</a></span><span class="t"><span class="key">def</span> <span class="nam">custom_generate_larger_households</span><span class="op">(</span><span class="nam">size</span><span class="op">,</span> <span class="nam">hh_sizes</span><span class="op">,</span> <span class="nam">hha_by_size_counts</span><span class="op">,</span> <span class="nam">hha_brackets</span><span class="op">,</span> <span class="nam">age_brackets</span><span class="op">,</span> <span class="nam">age_by_brackets_dic</span><span class="op">,</span> <span class="nam">contact_matrix_dic</span><span class="op">,</span> <span class="nam">single_year_age_distr</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t65" class="pln"><span class="n"><a href="#t65">65</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
<p id="t66" class="pln"><span class="n"><a href="#t66">66</a></span><span class="t"><span class="str"> Generate ages of those living in households of greater than one individual. Reference individual is sampled conditional on the household size.</span> </span><span class="r"></span></p>
<p id="t67" class="pln"><span class="n"><a href="#t67">67</a></span><span class="t"><span class="str"> All other household members have their ages sampled conditional on the reference person's age and the age mixing contact matrix</span> </span><span class="r"></span></p>
<p id="t68" class="pln"><span class="n"><a href="#t68">68</a></span><span class="t"><span class="str"> in households for the population under study.</span> </span><span class="r"></span></p>
<p id="t69" class="pln"><span class="n"><a href="#t69">69</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t70" class="pln"><span class="n"><a href="#t70">70</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
<p id="t71" class="pln"><span class="n"><a href="#t71">71</a></span><span class="t"><span class="str"> size (int) : The household size.</span> </span><span class="r"></span></p>
<p id="t72" class="pln"><span class="n"><a href="#t72">72</a></span><span class="t"><span class="str"> hh_sizes (array) : The count of household size s at index s-1.</span> </span><span class="r"></span></p>
<p id="t73" class="pln"><span class="n"><a href="#t73">73</a></span><span class="t"><span class="str"> hha_by_size_counts (matrix) : A matrix in which each row contains the age distribution of the reference person for household size s at index s-1.</span> </span><span class="r"></span></p>
<p id="t74" class="pln"><span class="n"><a href="#t74">74</a></span><span class="t"><span class="str"> hha_brackets (dict) : The age brackets for the heads of household.</span> </span><span class="r"></span></p>
<p id="t75" class="pln"><span class="n"><a href="#t75">75</a></span><span class="t"><span class="str"> age_brackets (dict) : A dictionary mapping age bracket keys to age bracket range.</span> </span><span class="r"></span></p>
<p id="t76" class="pln"><span class="n"><a href="#t76">76</a></span><span class="t"><span class="str"> age_by_brackets_dic (dict) : A dictionary mapping age to the age bracket range it falls within.</span> </span><span class="r"></span></p>
<p id="t77" class="pln"><span class="n"><a href="#t77">77</a></span><span class="t"><span class="str"> contact_matrix_dic (dict) : A dictionary of the age-specific contact matrix for different physical contact settings.</span> </span><span class="r"></span></p>
<p id="t78" class="pln"><span class="n"><a href="#t78">78</a></span><span class="t"><span class="str"> single_year_age_distr (dict) : The age distribution.</span> </span><span class="r"></span></p>
<p id="t79" class="pln"><span class="n"><a href="#t79">79</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t80" class="pln"><span class="n"><a href="#t80">80</a></span><span class="t"><span class="str"> Returns:</span> </span><span class="r"></span></p>
<p id="t81" class="pln"><span class="n"><a href="#t81">81</a></span><span class="t"><span class="str"> An array of households for size ``size`` where each household is a row and the values in the row are the ages of the household members.</span> </span><span class="r"></span></p>
<p id="t82" class="pln"><span class="n"><a href="#t82">82</a></span><span class="t"><span class="str"> The first age in the row is the age of the reference individual.</span> </span><span class="r"></span></p>
<p id="t83" class="pln"><span class="n"><a href="#t83">83</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
<p id="t84" class="run"><span class="n"><a href="#t84">84</a></span><span class="t"> <span class="nam">ya_coin</span> <span class="op">=</span> <span class="num">0.15</span> <span class="com"># This is a placeholder value. Users will need to change to fit whatever population you are working with</span> </span><span class="r"></span></p>
<p id="t85" class="pln"><span class="n"><a href="#t85">85</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t86" class="run"><span class="n"><a href="#t86">86</a></span><span class="t"> <span class="nam">homes</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">zeros</span><span class="op">(</span><span class="op">(</span><span class="nam">hh_sizes</span><span class="op">[</span><span class="nam">size</span><span class="op">-</span><span class="num">1</span><span class="op">]</span><span class="op">,</span> <span class="nam">size</span><span class="op">)</span><span class="op">,</span> <span class="nam">dtype</span><span class="op">=</span><span class="nam">int</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t87" class="pln"><span class="n"><a href="#t87">87</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t88" class="run"><span class="n"><a href="#t88">88</a></span><span class="t"> <span class="key">for</span> <span class="nam">h</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="nam">hh_sizes</span><span class="op">[</span><span class="nam">size</span><span class="op">-</span><span class="num">1</span><span class="op">]</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t89" class="pln"><span class="n"><a href="#t89">89</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t90" class="run"><span class="n"><a href="#t90">90</a></span><span class="t"> <span class="nam">hha</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">generate_household_head_age_by_size</span><span class="op">(</span><span class="nam">hha_by_size_counts</span><span class="op">,</span> <span class="nam">hha_brackets</span><span class="op">,</span> <span class="nam">size</span><span class="op">,</span> <span class="nam">single_year_age_distr</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t91" class="pln"><span class="n"><a href="#t91">91</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t92" class="run"><span class="n"><a href="#t92">92</a></span><span class="t"> <span class="nam">homes</span><span class="op">[</span><span class="nam">h</span><span class="op">]</span><span class="op">[</span><span class="num">0</span><span class="op">]</span> <span class="op">=</span> <span class="nam">hha</span> </span><span class="r"></span></p>
<p id="t93" class="pln"><span class="n"><a href="#t93">93</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t94" class="run"><span class="n"><a href="#t94">94</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="nam">age_by_brackets_dic</span><span class="op">[</span><span class="nam">hha</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t95" class="run"><span class="n"><a href="#t95">95</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="nam">min</span><span class="op">(</span><span class="nam">b</span><span class="op">,</span> <span class="nam">contact_matrix_dic</span><span class="op">[</span><span class="str">'H'</span><span class="op">]</span><span class="op">.</span><span class="nam">shape</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="op">-</span><span class="num">1</span><span class="op">)</span> <span class="com"># Ensure it doesn't go past the end of the array</span> </span><span class="r"></span></p>
<p id="t96" class="run"><span class="n"><a href="#t96">96</a></span><span class="t"> <span class="nam">b_prob</span> <span class="op">=</span> <span class="nam">contact_matrix_dic</span><span class="op">[</span><span class="str">'H'</span><span class="op">]</span><span class="op">[</span><span class="nam">b</span><span class="op">,</span> <span class="op">:</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t97" class="pln"><span class="n"><a href="#t97">97</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t98" class="run"><span class="n"><a href="#t98">98</a></span><span class="t"> <span class="key">for</span> <span class="nam">n</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">size</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t99" class="run"><span class="n"><a href="#t99">99</a></span><span class="t"> <span class="nam">bi</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">sample_single_arr</span><span class="op">(</span><span class="nam">b_prob</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t100" class="run"><span class="n"><a href="#t100">100</a></span><span class="t"> <span class="nam">ai</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">sample_from_range</span><span class="op">(</span><span class="nam">single_year_age_distr</span><span class="op">,</span> <span class="nam">age_brackets</span><span class="op">[</span><span class="nam">bi</span><span class="op">]</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="op">,</span> <span class="nam">age_brackets</span><span class="op">[</span><span class="nam">bi</span><span class="op">]</span><span class="op">[</span><span class="op">-</span><span class="num">1</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t101" class="pln"><span class="n"><a href="#t101">101</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t102" class="pln"><span class="n"><a href="#t102">102</a></span><span class="t"> <span class="str">""" The following is an example of how you may resample from an age range that is over produced and instead</span> </span><span class="r"></span></p>
<p id="t103" class="pln"><span class="n"><a href="#t103">103</a></span><span class="t"><span class="str"> sample ages from an age range that is under produced in your population. This kind of customization may</span> </span><span class="r"></span></p>
<p id="t104" class="pln"><span class="n"><a href="#t104">104</a></span><span class="t"><span class="str"> be necessary when your age mixing matrix and the population you are interested in modeling differ in</span> </span><span class="r"></span></p>
<p id="t105" class="pln"><span class="n"><a href="#t105">105</a></span><span class="t"><span class="str"> important but subtle ways. For example, generally household age mixing matrices reflect mixing patterns</span> </span><span class="r"></span></p>
<p id="t106" class="pln"><span class="n"><a href="#t106">106</a></span><span class="t"><span class="str"> for households composed of families. This means household age mixing matrices do not generally cover</span> </span><span class="r"></span></p>
<p id="t107" class="pln"><span class="n"><a href="#t107">107</a></span><span class="t"><span class="str"> college or university aged individuals living together. Without this customization, this algorithm tends</span> </span><span class="r"></span></p>
<p id="t108" class="pln"><span class="n"><a href="#t108">108</a></span><span class="t"><span class="str"> to under produce young adults. This method also has a tendency to underproduce the elderly, and does not</span> </span><span class="r"></span></p>
<p id="t109" class="pln"><span class="n"><a href="#t109">109</a></span><span class="t"><span class="str"> explicitly model the elderly living in nursing homes. Customizations like this should be considered in</span> </span><span class="r"></span></p>
<p id="t110" class="pln"><span class="n"><a href="#t110">110</a></span><span class="t"><span class="str"> context of the specific population and culture you are trying to model. In some cultures, it is common to</span> </span><span class="r"></span></p>
<p id="t111" class="pln"><span class="n"><a href="#t111">111</a></span><span class="t"><span class="str"> live in non-family households, while in others family households are the most common and include</span> </span><span class="r"></span></p>
<p id="t112" class="pln"><span class="n"><a href="#t112">112</a></span><span class="t"><span class="str"> multi-generational family households. If you are unsure of how to proceed with customizations please</span> </span><span class="r"></span></p>
<p id="t113" class="pln"><span class="n"><a href="#t113">113</a></span><span class="t"><span class="str"> take a look at the references listed in the overview documentation for more information.</span> </span><span class="r"></span></p>
<p id="t114" class="pln"><span class="n"><a href="#t114">114</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
<p id="t115" class="run"><span class="n"><a href="#t115">115</a></span><span class="t"> <span class="key">if</span> <span class="nam">ai</span> <span class="op">></span> <span class="num">5</span> <span class="key">and</span> <span class="nam">ai</span> <span class="op"><=</span> <span class="num">20</span><span class="op">:</span> <span class="com"># This a placeholder range. Users will need to change to fit whatever population you are working with</span> </span><span class="r"></span></p>
<p id="t116" class="run"><span class="n"><a href="#t116">116</a></span><span class="t"> <span class="key">if</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">binomial</span><span class="op">(</span><span class="num">1</span><span class="op">,</span> <span class="nam">ya_coin</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t117" class="run"><span class="n"><a href="#t117">117</a></span><span class="t"> <span class="nam">ai</span> <span class="op">=</span> <span class="nam">spsamp</span><span class="op">.</span><span class="nam">sample_from_range</span><span class="op">(</span><span class="nam">single_year_age_distr</span><span class="op">,</span> <span class="num">25</span><span class="op">,</span> <span class="num">32</span><span class="op">)</span> <span class="com"># This is a placeholder range. Users will need to change to fit whatever populaton you are working with</span> </span><span class="r"></span></p>
<p id="t118" class="pln"><span class="n"><a href="#t118">118</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t119" class="pln"><span class="n"><a href="#t119">119</a></span><span class="t"> <span class="com"># ai = spsamp.resample_age(single_year_age_distr, ai)</span> </span><span class="r"></span></p>
<p id="t120" class="run"><span class="n"><a href="#t120">120</a></span><span class="t"> <span class="nam">ai</span> <span class="op">=</span> <span class="nam">custom_resample_age</span><span class="op">(</span><span class="nam">single_year_age_distr</span><span class="op">,</span> <span class="nam">ai</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t121" class="pln"><span class="n"><a href="#t121">121</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t122" class="run"><span class="n"><a href="#t122">122</a></span><span class="t"> <span class="nam">homes</span><span class="op">[</span><span class="nam">h</span><span class="op">]</span><span class="op">[</span><span class="nam">n</span><span class="op">]</span> <span class="op">=</span> <span class="nam">ai</span> </span><span class="r"></span></p>
<p id="t123" class="pln"><span class="n"><a href="#t123">123</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t124" class="run"><span class="n"><a href="#t124">124</a></span><span class="t"> <span class="key">return</span> <span class="nam">homes</span> </span><span class="r"></span></p>
<p id="t125" class="pln"><span class="n"><a href="#t125">125</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t126" class="pln"><span class="n"><a href="#t126">126</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t127" class="run"><span class="n"><a href="#t127">127</a></span><span class="t"><span class="key">def</span> <span class="nam">custom_generate_all_households</span><span class="op">(</span><span class="nam">N</span><span class="op">,</span> <span class="nam">hh_sizes</span><span class="op">,</span> <span class="nam">hha_by_size_counts</span><span class="op">,</span> <span class="nam">hha_brackets</span><span class="op">,</span> <span class="nam">age_brackets</span><span class="op">,</span> <span class="nam">age_by_brackets_dic</span><span class="op">,</span> <span class="nam">contact_matrix_dic</span><span class="op">,</span> <span class="nam">single_year_age_distr</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t128" class="pln"><span class="n"><a href="#t128">128</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
<p id="t129" class="pln"><span class="n"><a href="#t129">129</a></span><span class="t"><span class="str"> Generate the ages of those living in households together. First create households of people living alone, then larger households.</span> </span><span class="r"></span></p>
<p id="t130" class="pln"><span class="n"><a href="#t130">130</a></span><span class="t"><span class="str"> For households larger than 1, a reference individual's age is sampled conditional on the household size, while all other household</span> </span><span class="r"></span></p>
<p id="t131" class="pln"><span class="n"><a href="#t131">131</a></span><span class="t"><span class="str"> members have their ages sampled conditional on the reference person's age and the age mixing contact matrix in households</span> </span><span class="r"></span></p>
<p id="t132" class="pln"><span class="n"><a href="#t132">132</a></span><span class="t"><span class="str"> for the population under study.</span> </span><span class="r"></span></p>
<p id="t133" class="pln"><span class="n"><a href="#t133">133</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t134" class="pln"><span class="n"><a href="#t134">134</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
<p id="t135" class="pln"><span class="n"><a href="#t135">135</a></span><span class="t"><span class="str"> N (int) : The number of people in the population.</span> </span><span class="r"></span></p>
<p id="t136" class="pln"><span class="n"><a href="#t136">136</a></span><span class="t"><span class="str"> hh_sizes (array) : The count of household size s at index s-1.</span> </span><span class="r"></span></p>
<p id="t137" class="pln"><span class="n"><a href="#t137">137</a></span><span class="t"><span class="str"> hha_by_size_counts (matrix) : A matrix in which each row contains the age distribution of the reference person for household size s at index s-1.</span> </span><span class="r"></span></p>
<p id="t138" class="pln"><span class="n"><a href="#t138">138</a></span><span class="t"><span class="str"> hha_brackets (dict) : The age brackets for the heads of household.</span> </span><span class="r"></span></p>
<p id="t139" class="pln"><span class="n"><a href="#t139">139</a></span><span class="t"><span class="str"> age_brackets (dict) : The dictionary mapping age bracket keys to age bracket range.</span> </span><span class="r"></span></p>
<p id="t140" class="pln"><span class="n"><a href="#t140">140</a></span><span class="t"><span class="str"> age_by_brackets_dic (dict) : The dictionary mapping age to the age bracket range it falls within.</span> </span><span class="r"></span></p>
<p id="t141" class="pln"><span class="n"><a href="#t141">141</a></span><span class="t"><span class="str"> contact_matrix_dic (dict) : The dictionary of the age-specific contact matrix for different physical contact settings.</span> </span><span class="r"></span></p>
<p id="t142" class="pln"><span class="n"><a href="#t142">142</a></span><span class="t"><span class="str"> single_year_age_distr (dict) : The age distribution.</span> </span><span class="r"></span></p>
<p id="t143" class="pln"><span class="n"><a href="#t143">143</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t144" class="pln"><span class="n"><a href="#t144">144</a></span><span class="t"><span class="str"> Returns:</span> </span><span class="r"></span></p>
<p id="t145" class="pln"><span class="n"><a href="#t145">145</a></span><span class="t"><span class="str"> An array of all households where each household is a row and the values in the row are the ages of the household members.</span> </span><span class="r"></span></p>
<p id="t146" class="pln"><span class="n"><a href="#t146">146</a></span><span class="t"><span class="str"> The first age in the row is the age of the reference individual. Households are randomly shuffled by size.</span> </span><span class="r"></span></p>
<p id="t147" class="pln"><span class="n"><a href="#t147">147</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
<p id="t148" class="pln"><span class="n"><a href="#t148">148</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t149" class="run"><span class="n"><a href="#t149">149</a></span><span class="t"> <span class="nam">homes_dic</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t150" class="run"><span class="n"><a href="#t150">150</a></span><span class="t"> <span class="nam">homes_dic</span><span class="op">[</span><span class="num">1</span><span class="op">]</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">generate_living_alone</span><span class="op">(</span><span class="nam">hh_sizes</span><span class="op">,</span> <span class="nam">hha_by_size_counts</span><span class="op">,</span> <span class="nam">hha_brackets</span><span class="op">,</span> <span class="nam">single_year_age_distr</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t151" class="pln"><span class="n"><a href="#t151">151</a></span><span class="t"> <span class="com"># remove living alone from the distribution to choose from!</span> </span><span class="r"></span></p>
<p id="t152" class="run"><span class="n"><a href="#t152">152</a></span><span class="t"> <span class="key">for</span> <span class="nam">h</span> <span class="key">in</span> <span class="nam">homes_dic</span><span class="op">[</span><span class="num">1</span><span class="op">]</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t153" class="run"><span class="n"><a href="#t153">153</a></span><span class="t"> <span class="nam">single_year_age_distr</span><span class="op">[</span><span class="nam">h</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="op">]</span> <span class="op">-=</span> <span class="num">1.0</span><span class="op">/</span><span class="nam">N</span> </span><span class="r"></span></p>
<p id="t154" class="pln"><span class="n"><a href="#t154">154</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t155" class="pln"><span class="n"><a href="#t155">155</a></span><span class="t"> <span class="com"># generate larger households and the ages of people living in them</span> </span><span class="r"></span></p>
<p id="t156" class="run"><span class="n"><a href="#t156">156</a></span><span class="t"> <span class="key">for</span> <span class="nam">s</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="num">2</span><span class="op">,</span> <span class="num">8</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t157" class="run"><span class="n"><a href="#t157">157</a></span><span class="t"> <span class="nam">homes_dic</span><span class="op">[</span><span class="nam">s</span><span class="op">]</span> <span class="op">=</span> <span class="nam">custom_generate_larger_households</span><span class="op">(</span><span class="nam">s</span><span class="op">,</span> <span class="nam">hh_sizes</span><span class="op">,</span> <span class="nam">hha_by_size_counts</span><span class="op">,</span> <span class="nam">hha_brackets</span><span class="op">,</span> <span class="nam">age_brackets</span><span class="op">,</span> <span class="nam">age_by_brackets_dic</span><span class="op">,</span> <span class="nam">contact_matrix_dic</span><span class="op">,</span> <span class="nam">single_year_age_distr</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t158" class="pln"><span class="n"><a href="#t158">158</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t159" class="run"><span class="n"><a href="#t159">159</a></span><span class="t"> <span class="nam">homes</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t160" class="run"><span class="n"><a href="#t160">160</a></span><span class="t"> <span class="key">for</span> <span class="nam">s</span> <span class="key">in</span> <span class="nam">homes_dic</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t161" class="run"><span class="n"><a href="#t161">161</a></span><span class="t"> <span class="nam">homes</span> <span class="op">+=</span> <span class="nam">list</span><span class="op">(</span><span class="nam">homes_dic</span><span class="op">[</span><span class="nam">s</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t162" class="pln"><span class="n"><a href="#t162">162</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t163" class="run"><span class="n"><a href="#t163">163</a></span><span class="t"> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">shuffle</span><span class="op">(</span><span class="nam">homes</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t164" class="run"><span class="n"><a href="#t164">164</a></span><span class="t"> <span class="key">return</span> <span class="nam">homes_dic</span><span class="op">,</span> <span class="nam">homes</span> </span><span class="r"></span></p>
<p id="t165" class="pln"><span class="n"><a href="#t165">165</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t166" class="pln"><span class="n"><a href="#t166">166</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t167" class="run"><span class="n"><a href="#t167">167</a></span><span class="t"><span class="key">def</span> <span class="nam">generate_microstructure_with_facilities</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">n</span><span class="op">,</span> <span class="nam">sheet_name</span><span class="op">=</span><span class="str">'United States of America'</span><span class="op">,</span> <span class="nam">school_enrollment_counts_available</span><span class="op">=</span><span class="key">True</span><span class="op">,</span> <span class="nam">plot</span><span class="op">=</span><span class="key">False</span><span class="op">,</span> <span class="nam">verbose</span><span class="op">=</span><span class="key">False</span><span class="op">,</span> <span class="nam">write</span><span class="op">=</span><span class="key">False</span><span class="op">,</span> <span class="nam">return_popdict</span><span class="op">=</span><span class="key">False</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="key">False</span><span class="op">,</span> <span class="nam">use_two_group_reduction</span><span class="op">=</span><span class="key">False</span><span class="op">,</span> <span class="nam">average_LTCF_degree</span><span class="op">=</span><span class="num">20</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t168" class="pln"><span class="n"><a href="#t168">168</a></span><span class="t"> <span class="com"># Use part 2 of the Long Term Care Facilities</span> </span><span class="r"></span></p>
<p id="t169" class="run"><span class="n"><a href="#t169">169</a></span><span class="t"> <span class="nam">part</span> <span class="op">=</span> <span class="num">2</span> </span><span class="r"></span></p>
<p id="t170" class="pln"><span class="n"><a href="#t170">170</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t171" class="pln"><span class="n"><a href="#t171">171</a></span><span class="t"> <span class="com"># Grab Long Term Care Facilities data</span> </span><span class="r"></span></p>
<p id="t172" class="run"><span class="n"><a href="#t172">172</a></span><span class="t"> <span class="nam">ltcf_df</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_usa_long_term_care_facility_data</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">part</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t173" class="pln"><span class="n"><a href="#t173">173</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t174" class="pln"><span class="n"><a href="#t174">174</a></span><span class="t"> <span class="com"># ltcf_df keys</span> </span><span class="r"></span></p>
<p id="t175" class="run"><span class="n"><a href="#t175">175</a></span><span class="t"> <span class="nam">ltcf_age_bracket_keys</span> <span class="op">=</span> <span class="op">[</span><span class="str">'Under 65'</span><span class="op">,</span> <span class="str">'65–74'</span><span class="op">,</span> <span class="str">'75–84'</span><span class="op">,</span> <span class="str">'85 and over'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t176" class="run"><span class="n"><a href="#t176">176</a></span><span class="t"> <span class="nam">facility_keys</span> <span class="op">=</span> <span class="op">[</span> </span><span class="r"></span></p>
<p id="t177" class="pln"><span class="n"><a href="#t177">177</a></span><span class="t"> <span class="com"># 'Hospice',</span> </span><span class="r"></span></p>
<p id="t178" class="pln"><span class="n"><a href="#t178">178</a></span><span class="t"> <span class="str">'Nursing home'</span><span class="op">,</span> </span><span class="r"></span></p>
<p id="t179" class="pln"><span class="n"><a href="#t179">179</a></span><span class="t"> <span class="str">'Residential care community'</span> </span><span class="r"></span></p>
<p id="t180" class="pln"><span class="n"><a href="#t180">180</a></span><span class="t"> <span class="op">]</span> </span><span class="r"></span></p>
<p id="t181" class="pln"><span class="n"><a href="#t181">181</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t182" class="pln"><span class="n"><a href="#t182">182</a></span><span class="t"> <span class="com"># state numbers</span> </span><span class="r"></span></p>
<p id="t183" class="run"><span class="n"><a href="#t183">183</a></span><span class="t"> <span class="nam">facillity_users</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t184" class="run"><span class="n"><a href="#t184">184</a></span><span class="t"> <span class="key">for</span> <span class="nam">fk</span> <span class="key">in</span> <span class="nam">facility_keys</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t185" class="run"><span class="n"><a href="#t185">185</a></span><span class="t"> <span class="nam">facillity_users</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t186" class="run"><span class="n"><a href="#t186">186</a></span><span class="t"> <span class="nam">facillity_users</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">[</span><span class="str">'Total'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">ltcf_df</span><span class="op">[</span><span class="nam">ltcf_df</span><span class="op">.</span><span class="nam">iloc</span><span class="op">[</span><span class="op">:</span><span class="op">,</span> <span class="num">0</span><span class="op">]</span> <span class="op">==</span> <span class="str">'Number of users2, 5'</span><span class="op">]</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">.</span><span class="nam">values</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="op">.</span><span class="nam">replace</span><span class="op">(</span><span class="str">','</span><span class="op">,</span> <span class="str">''</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t187" class="run"><span class="n"><a href="#t187">187</a></span><span class="t"> <span class="key">for</span> <span class="nam">ab</span> <span class="key">in</span> <span class="nam">ltcf_age_bracket_keys</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t188" class="run"><span class="n"><a href="#t188">188</a></span><span class="t"> <span class="nam">facillity_users</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">[</span><span class="nam">ab</span><span class="op">]</span> <span class="op">=</span> <span class="nam">float</span><span class="op">(</span><span class="nam">ltcf_df</span><span class="op">[</span><span class="nam">ltcf_df</span><span class="op">.</span><span class="nam">iloc</span><span class="op">[</span><span class="op">:</span><span class="op">,</span> <span class="num">0</span><span class="op">]</span> <span class="op">==</span> <span class="nam">ab</span><span class="op">]</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">.</span><span class="nam">values</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="op">.</span><span class="nam">replace</span><span class="op">(</span><span class="str">','</span><span class="op">,</span> <span class="str">''</span><span class="op">)</span><span class="op">)</span><span class="op">/</span><span class="num">100.</span> </span><span class="r"></span></p>
<p id="t189" class="pln"><span class="n"><a href="#t189">189</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t190" class="run"><span class="n"><a href="#t190">190</a></span><span class="t"> <span class="nam">total_facility_users</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">sum</span><span class="op">(</span><span class="op">[</span><span class="nam">facillity_users</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">[</span><span class="str">'Total'</span><span class="op">]</span> <span class="key">for</span> <span class="nam">fk</span> <span class="key">in</span> <span class="nam">facillity_users</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t191" class="pln"><span class="n"><a href="#t191">191</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t192" class="pln"><span class="n"><a href="#t192">192</a></span><span class="t"> <span class="com"># Census Bureau numbers 2016</span> </span><span class="r"></span></p>
<p id="t193" class="run"><span class="n"><a href="#t193">193</a></span><span class="t"> <span class="nam">state_pop_2016</span> <span class="op">=</span> <span class="num">7288000</span> </span><span class="r"></span></p>
<p id="t194" class="run"><span class="n"><a href="#t194">194</a></span><span class="t"> <span class="nam">state_age_distr_2016</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t195" class="run"><span class="n"><a href="#t195">195</a></span><span class="t"> <span class="nam">state_age_distr_2016</span><span class="op">[</span><span class="str">'60-64'</span><span class="op">]</span> <span class="op">=</span> <span class="num">6.3</span> </span><span class="r"></span></p>
<p id="t196" class="run"><span class="n"><a href="#t196">196</a></span><span class="t"> <span class="nam">state_age_distr_2016</span><span class="op">[</span><span class="str">'65-74'</span><span class="op">]</span> <span class="op">=</span> <span class="num">9.0</span> </span><span class="r"></span></p>
<p id="t197" class="run"><span class="n"><a href="#t197">197</a></span><span class="t"> <span class="nam">state_age_distr_2016</span><span class="op">[</span><span class="str">'75-84'</span><span class="op">]</span> <span class="op">=</span> <span class="num">4.0</span> </span><span class="r"></span></p>
<p id="t198" class="run"><span class="n"><a href="#t198">198</a></span><span class="t"> <span class="nam">state_age_distr_2016</span><span class="op">[</span><span class="str">'85-100'</span><span class="op">]</span> <span class="op">=</span> <span class="num">1.8</span> </span><span class="r"></span></p>
<p id="t199" class="pln"><span class="n"><a href="#t199">199</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t200" class="pln"><span class="n"><a href="#t200">200</a></span><span class="t"> <span class="com"># Census Bureau numbers 2018</span> </span><span class="r"></span></p>
<p id="t201" class="run"><span class="n"><a href="#t201">201</a></span><span class="t"> <span class="nam">state_pop_2018</span> <span class="op">=</span> <span class="num">7535591</span> </span><span class="r"></span></p>
<p id="t202" class="run"><span class="n"><a href="#t202">202</a></span><span class="t"> <span class="nam">state_age_distr_2018</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t203" class="run"><span class="n"><a href="#t203">203</a></span><span class="t"> <span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="str">'60-64'</span><span class="op">]</span> <span class="op">=</span> <span class="num">6.3</span> </span><span class="r"></span></p>
<p id="t204" class="run"><span class="n"><a href="#t204">204</a></span><span class="t"> <span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="str">'65-74'</span><span class="op">]</span> <span class="op">=</span> <span class="num">9.5</span> </span><span class="r"></span></p>
<p id="t205" class="run"><span class="n"><a href="#t205">205</a></span><span class="t"> <span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="str">'75-84'</span><span class="op">]</span> <span class="op">=</span> <span class="num">4.3</span> </span><span class="r"></span></p>
<p id="t206" class="run"><span class="n"><a href="#t206">206</a></span><span class="t"> <span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="str">'85-100'</span><span class="op">]</span> <span class="op">=</span> <span class="num">1.8</span> </span><span class="r"></span></p>
<p id="t207" class="pln"><span class="n"><a href="#t207">207</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t208" class="run"><span class="n"><a href="#t208">208</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">state_age_distr_2016</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t209" class="run"><span class="n"><a href="#t209">209</a></span><span class="t"> <span class="nam">state_age_distr_2016</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">state_age_distr_2016</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">/</span><span class="num">100.</span> </span><span class="r"></span></p>
<p id="t210" class="run"><span class="n"><a href="#t210">210</a></span><span class="t"> <span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">/</span><span class="num">100.</span> </span><span class="r"></span></p>
<p id="t211" class="pln"><span class="n"><a href="#t211">211</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t212" class="run"><span class="n"><a href="#t212">212</a></span><span class="t"> <span class="nam">num_state_elderly_2016</span> <span class="op">=</span> <span class="num">0</span> </span><span class="r"></span></p>
<p id="t213" class="run"><span class="n"><a href="#t213">213</a></span><span class="t"> <span class="nam">num_state_elderly_2018</span> <span class="op">=</span> <span class="num">0</span> </span><span class="r"></span></p>
<p id="t214" class="run"><span class="n"><a href="#t214">214</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">state_age_distr_2016</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t215" class="run"><span class="n"><a href="#t215">215</a></span><span class="t"> <span class="nam">num_state_elderly_2016</span> <span class="op">+=</span> <span class="nam">state_pop_2016</span> <span class="op">*</span> <span class="nam">state_age_distr_2016</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t216" class="run"><span class="n"><a href="#t216">216</a></span><span class="t"> <span class="nam">num_state_elderly_2018</span> <span class="op">+=</span> <span class="nam">state_pop_2018</span> <span class="op">*</span> <span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t217" class="pln"><span class="n"><a href="#t217">217</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t218" class="run"><span class="n"><a href="#t218">218</a></span><span class="t"> <span class="nam">expected_users_2018</span> <span class="op">=</span> <span class="nam">total_facility_users</span> <span class="op">*</span> <span class="nam">num_state_elderly_2018</span><span class="op">/</span><span class="nam">num_state_elderly_2016</span> </span><span class="r"></span></p>
<p id="t219" class="pln"><span class="n"><a href="#t219">219</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t220" class="run"><span class="n"><a href="#t220">220</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t221" class="mis show_mis"><span class="n"><a href="#t221">221</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'number of elderly'</span><span class="op">,</span> <span class="nam">num_state_elderly_2016</span><span class="op">,</span> <span class="nam">num_state_elderly_2018</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t222" class="mis show_mis"><span class="n"><a href="#t222">222</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'growth in elderly'</span><span class="op">,</span> <span class="nam">num_state_elderly_2018</span><span class="op">/</span><span class="nam">num_state_elderly_2016</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t223" class="mis show_mis"><span class="n"><a href="#t223">223</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'users in 2016'</span><span class="op">,</span> <span class="nam">total_facility_users</span><span class="op">,</span> <span class="str">'% of elderly'</span><span class="op">,</span> <span class="nam">total_facility_users</span><span class="op">/</span><span class="nam">num_state_elderly_2016</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t224" class="mis show_mis"><span class="n"><a href="#t224">224</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'users in 2018'</span><span class="op">,</span> <span class="nam">expected_users_2018</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t225" class="pln"><span class="n"><a href="#t225">225</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t226" class="pln"><span class="n"><a href="#t226">226</a></span><span class="t"> <span class="com"># location age distribution</span> </span><span class="r"></span></p>
<p id="t227" class="run"><span class="n"><a href="#t227">227</a></span><span class="t"> <span class="nam">age_distr_16</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">read_age_bracket_distr</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t228" class="run"><span class="n"><a href="#t228">228</a></span><span class="t"> <span class="nam">age_brackets_16</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_census_age_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t229" class="run"><span class="n"><a href="#t229">229</a></span><span class="t"> <span class="nam">age_by_brackets_dic_16</span> <span class="op">=</span> <span class="nam">spb</span><span class="op">.</span><span class="nam">get_age_by_brackets_dic</span><span class="op">(</span><span class="nam">age_brackets_16</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t230" class="pln"><span class="n"><a href="#t230">230</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t231" class="pln"><span class="n"><a href="#t231">231</a></span><span class="t"> <span class="com"># current King County population size</span> </span><span class="r"></span></p>
<p id="t232" class="run"><span class="n"><a href="#t232">232</a></span><span class="t"> <span class="nam">pop</span> <span class="op">=</span> <span class="num">2.25e6</span> </span><span class="r"></span></p>
<p id="t233" class="pln"><span class="n"><a href="#t233">233</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t234" class="pln"><span class="n"><a href="#t234">234</a></span><span class="t"> <span class="com"># local elderly population estimate</span> </span><span class="r"></span></p>
<p id="t235" class="run"><span class="n"><a href="#t235">235</a></span><span class="t"> <span class="nam">local_elderly_2018</span> <span class="op">=</span> <span class="num">0</span> </span><span class="r"></span></p>
<p id="t236" class="run"><span class="n"><a href="#t236">236</a></span><span class="t"> <span class="key">for</span> <span class="nam">ab</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="num">12</span><span class="op">,</span> <span class="num">16</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t237" class="run"><span class="n"><a href="#t237">237</a></span><span class="t"> <span class="nam">local_elderly_2018</span> <span class="op">+=</span> <span class="nam">age_distr_16</span><span class="op">[</span><span class="nam">ab</span><span class="op">]</span> <span class="op">*</span> <span class="nam">pop</span> </span><span class="r"></span></p>
<p id="t238" class="pln"><span class="n"><a href="#t238">238</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t239" class="run"><span class="n"><a href="#t239">239</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t240" class="mis show_mis"><span class="n"><a href="#t240">240</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'number of local elderly'</span><span class="op">,</span> <span class="nam">local_elderly_2018</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t241" class="pln"><span class="n"><a href="#t241">241</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t242" class="pln"><span class="n"><a href="#t242">242</a></span><span class="t"> <span class="com"># growth_since_2016 = num_state_elderly_2018/num_state_elderly_2016</span> </span><span class="r"></span></p>
<p id="t243" class="pln"><span class="n"><a href="#t243">243</a></span><span class="t"> <span class="com"># local_perc_elderly_2018 = local_elderly_2018/num_state_elderly_2018</span> </span><span class="r"></span></p>
<p id="t244" class="pln"><span class="n"><a href="#t244">244</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t245" class="run"><span class="n"><a href="#t245">245</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t246" class="mis show_mis"><span class="n"><a href="#t246">246</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'local users in 2018?'</span><span class="op">,</span> <span class="nam">total_facility_users</span> <span class="op">*</span> <span class="nam">local_elderly_2018</span><span class="op">/</span><span class="nam">num_state_elderly_2018</span> <span class="op">*</span> <span class="nam">num_state_elderly_2018</span><span class="op">/</span><span class="nam">num_state_elderly_2016</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t247" class="pln"><span class="n"><a href="#t247">247</a></span><span class="t"> <span class="com"># seattle_users_est_from_state = total_facility_users * local_perc_elderly_2018 * growth_since_2016</span> </span><span class="r"></span></p>
<p id="t248" class="pln"><span class="n"><a href="#t248">248</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t249" class="run"><span class="n"><a href="#t249">249</a></span><span class="t"> <span class="nam">est_seattle_users_2018</span> <span class="op">=</span> <span class="nam">dict</span><span class="op">.</span><span class="nam">fromkeys</span><span class="op">(</span><span class="op">[</span><span class="str">'60-64'</span><span class="op">,</span> <span class="str">'65-74'</span><span class="op">,</span> <span class="str">'75-84'</span><span class="op">,</span> <span class="str">'85-100'</span><span class="op">]</span><span class="op">,</span> <span class="num">0</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t250" class="pln"><span class="n"><a href="#t250">250</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t251" class="run"><span class="n"><a href="#t251">251</a></span><span class="t"> <span class="key">for</span> <span class="nam">fk</span> <span class="key">in</span> <span class="nam">facillity_users</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t252" class="run"><span class="n"><a href="#t252">252</a></span><span class="t"> <span class="key">for</span> <span class="nam">ab</span> <span class="key">in</span> <span class="nam">facillity_users</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t253" class="run"><span class="n"><a href="#t253">253</a></span><span class="t"> <span class="key">if</span> <span class="nam">ab</span> <span class="op">!=</span> <span class="str">'Total'</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t254" class="pln"><span class="n"><a href="#t254">254</a></span><span class="t"> <span class="com"># print(fk, ab, facillity_users[fk][ab], facillity_users[fk][ab] * facillity_users[fk]['Total'], facillity_users[fk][ab] * facillity_users[fk]['Total'] * pop/state_pop_2018)</span> </span><span class="r"></span></p>
<p id="t255" class="run"><span class="n"><a href="#t255">255</a></span><span class="t"> <span class="key">if</span> <span class="nam">ab</span> <span class="op">==</span> <span class="str">'Under 65'</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t256" class="run"><span class="n"><a href="#t256">256</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="str">'60-64'</span> </span><span class="r"></span></p>
<p id="t257" class="run"><span class="n"><a href="#t257">257</a></span><span class="t"> <span class="key">elif</span> <span class="nam">ab</span> <span class="op">==</span> <span class="str">'65–74'</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t258" class="run"><span class="n"><a href="#t258">258</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="str">'65-74'</span> </span><span class="r"></span></p>
<p id="t259" class="run"><span class="n"><a href="#t259">259</a></span><span class="t"> <span class="key">elif</span> <span class="nam">ab</span> <span class="op">==</span> <span class="str">'75–84'</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t260" class="run"><span class="n"><a href="#t260">260</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="str">'75-84'</span> </span><span class="r"></span></p>
<p id="t261" class="run"><span class="n"><a href="#t261">261</a></span><span class="t"> <span class="key">elif</span> <span class="nam">ab</span> <span class="op">==</span> <span class="str">'85 and over'</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t262" class="run"><span class="n"><a href="#t262">262</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="str">'85-100'</span> </span><span class="r"></span></p>
<p id="t263" class="run"><span class="n"><a href="#t263">263</a></span><span class="t"> <span class="nam">est_seattle_users_2018</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span> <span class="op">+=</span> <span class="nam">facillity_users</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">[</span><span class="nam">ab</span><span class="op">]</span> <span class="op">*</span> <span class="nam">facillity_users</span><span class="op">[</span><span class="nam">fk</span><span class="op">]</span><span class="op">[</span><span class="str">'Total'</span><span class="op">]</span> <span class="op">*</span> <span class="nam">pop</span><span class="op">/</span><span class="nam">state_pop_2018</span> </span><span class="r"></span></p>
<p id="t264" class="pln"><span class="n"><a href="#t264">264</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t265" class="run"><span class="n"><a href="#t265">265</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t266" class="mis show_mis"><span class="n"><a href="#t266">266</a></span><span class="t"> <span class="key">for</span> <span class="nam">ab</span> <span class="key">in</span> <span class="nam">est_seattle_users_2018</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t267" class="mis show_mis"><span class="n"><a href="#t267">267</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">ab</span><span class="op">,</span> <span class="nam">est_seattle_users_2018</span><span class="op">[</span><span class="nam">ab</span><span class="op">]</span><span class="op">,</span> <span class="nam">est_seattle_users_2018</span><span class="op">[</span><span class="nam">ab</span><span class="op">]</span><span class="op">/</span><span class="op">(</span><span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="nam">ab</span><span class="op">]</span> <span class="op">*</span> <span class="nam">pop</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t268" class="mis show_mis"><span class="n"><a href="#t268">268</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">np</span><span class="op">.</span><span class="nam">sum</span><span class="op">(</span><span class="op">[</span><span class="nam">est_seattle_users_2018</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span> <span class="key">for</span> <span class="nam">b</span> <span class="key">in</span> <span class="nam">est_seattle_users_2018</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t269" class="pln"><span class="n"><a href="#t269">269</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t270" class="pln"><span class="n"><a href="#t270">270</a></span><span class="t"> <span class="com"># for pop of 2.25 million of Seattle</span> </span><span class="r"></span></p>
<p id="t271" class="run"><span class="n"><a href="#t271">271</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t272" class="run"><span class="n"><a href="#t272">272</a></span><span class="t"> <span class="key">for</span> <span class="nam">b</span> <span class="key">in</span> <span class="nam">est_seattle_users_2018</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t273" class="run"><span class="n"><a href="#t273">273</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span> <span class="op">=</span> <span class="nam">est_seattle_users_2018</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">/</span><span class="nam">state_age_distr_2018</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">/</span><span class="nam">pop</span> </span><span class="r"></span></p>
<p id="t274" class="pln"><span class="n"><a href="#t274">274</a></span><span class="t"> <span class="com"># print(b,est_ltcf_user_by_age_brackets_perc[b])</span> </span><span class="r"></span></p>
<p id="t275" class="pln"><span class="n"><a href="#t275">275</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t276" class="run"><span class="n"><a href="#t276">276</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'65-69'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'65-74'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t277" class="run"><span class="n"><a href="#t277">277</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'70-74'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'65-74'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t278" class="run"><span class="n"><a href="#t278">278</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'75-79'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'75-84'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t279" class="run"><span class="n"><a href="#t279">279</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'80-84'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'75-84'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t280" class="pln"><span class="n"><a href="#t280">280</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t281" class="run"><span class="n"><a href="#t281">281</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">.</span><span class="nam">pop</span><span class="op">(</span><span class="str">'65-74'</span><span class="op">,</span> <span class="key">None</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t282" class="run"><span class="n"><a href="#t282">282</a></span><span class="t"> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">.</span><span class="nam">pop</span><span class="op">(</span><span class="str">'75-84'</span><span class="op">,</span> <span class="key">None</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t283" class="pln"><span class="n"><a href="#t283">283</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t284" class="run"><span class="n"><a href="#t284">284</a></span><span class="t"> <span class="nam">age_distr_18_fp</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="str">'demographics'</span><span class="op">,</span> <span class="str">'contact_matrices_152_countries'</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="str">'age_distributions'</span><span class="op">,</span> <span class="str">'seattle_metro_age_bracket_distr_18.dat'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t285" class="run"><span class="n"><a href="#t285">285</a></span><span class="t"> <span class="nam">age_distr_18</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">read_age_bracket_distr</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">file_path</span><span class="op">=</span><span class="nam">age_distr_18_fp</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t286" class="run"><span class="n"><a href="#t286">286</a></span><span class="t"> <span class="nam">age_brackets_18_fp</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="str">'demographics'</span><span class="op">,</span> <span class="str">'contact_matrices_152_countries'</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="str">'age_distributions'</span><span class="op">,</span> <span class="str">'census_age_brackets_18.dat'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t287" class="run"><span class="n"><a href="#t287">287</a></span><span class="t"> <span class="nam">age_brackets_18</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_census_age_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">file_path</span><span class="op">=</span><span class="nam">age_brackets_18_fp</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t288" class="run"><span class="n"><a href="#t288">288</a></span><span class="t"> <span class="nam">age_by_brackets_dic_18</span> <span class="op">=</span> <span class="nam">spb</span><span class="op">.</span><span class="nam">get_age_by_brackets_dic</span><span class="op">(</span><span class="nam">age_brackets_18</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t289" class="pln"><span class="n"><a href="#t289">289</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t290" class="pln"><span class="n"><a href="#t290">290</a></span><span class="t"> <span class="com"># subfolder where the population files will be saved</span> </span><span class="r"></span></p>
<p id="t291" class="run"><span class="n"><a href="#t291">291</a></span><span class="t"> <span class="nam">folder_name</span> <span class="op">=</span> <span class="str">'contact_networks_facilities'</span> </span><span class="r"></span></p>
<p id="t292" class="pln"><span class="n"><a href="#t292">292</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t293" class="run"><span class="n"><a href="#t293">293</a></span><span class="t"> <span class="nam">n</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">n</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t294" class="pln"><span class="n"><a href="#t294">294</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t295" class="run"><span class="n"><a href="#t295">295</a></span><span class="t"> <span class="nam">expected_users_by_age</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t296" class="pln"><span class="n"><a href="#t296">296</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t297" class="run"><span class="n"><a href="#t297">297</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="num">60</span><span class="op">,</span> <span class="num">101</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t298" class="run"><span class="n"><a href="#t298">298</a></span><span class="t"> <span class="key">if</span> <span class="nam">a</span> <span class="op"><</span> <span class="num">65</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t299" class="run"><span class="n"><a href="#t299">299</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="nam">age_by_brackets_dic_18</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t300" class="pln"><span class="n"><a href="#t300">300</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t301" class="run"><span class="n"><a href="#t301">301</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">n</span> <span class="op">*</span> <span class="nam">age_distr_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">/</span><span class="nam">len</span><span class="op">(</span><span class="nam">age_brackets_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t302" class="run"><span class="n"><a href="#t302">302</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">*</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'60-64'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t303" class="run"><span class="n"><a href="#t303">303</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">math</span><span class="op">.</span><span class="nam">ceil</span><span class="op">(</span><span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t304" class="pln"><span class="n"><a href="#t304">304</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t305" class="run"><span class="n"><a href="#t305">305</a></span><span class="t"> <span class="key">elif</span> <span class="nam">a</span> <span class="op"><</span> <span class="num">75</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t306" class="run"><span class="n"><a href="#t306">306</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="nam">age_by_brackets_dic_18</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t307" class="pln"><span class="n"><a href="#t307">307</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t308" class="run"><span class="n"><a href="#t308">308</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">n</span> <span class="op">*</span> <span class="nam">age_distr_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">/</span><span class="nam">len</span><span class="op">(</span><span class="nam">age_brackets_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t309" class="run"><span class="n"><a href="#t309">309</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">*</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'70-74'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t310" class="run"><span class="n"><a href="#t310">310</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">math</span><span class="op">.</span><span class="nam">ceil</span><span class="op">(</span><span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t311" class="pln"><span class="n"><a href="#t311">311</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t312" class="run"><span class="n"><a href="#t312">312</a></span><span class="t"> <span class="key">elif</span> <span class="nam">a</span> <span class="op"><</span> <span class="num">85</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t313" class="run"><span class="n"><a href="#t313">313</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="nam">age_by_brackets_dic_18</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t314" class="pln"><span class="n"><a href="#t314">314</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t315" class="run"><span class="n"><a href="#t315">315</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">n</span> <span class="op">*</span> <span class="nam">age_distr_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">/</span><span class="nam">len</span><span class="op">(</span><span class="nam">age_brackets_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t316" class="run"><span class="n"><a href="#t316">316</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">*</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'80-84'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t317" class="run"><span class="n"><a href="#t317">317</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">math</span><span class="op">.</span><span class="nam">ceil</span><span class="op">(</span><span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t318" class="pln"><span class="n"><a href="#t318">318</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t319" class="run"><span class="n"><a href="#t319">319</a></span><span class="t"> <span class="key">elif</span> <span class="nam">a</span> <span class="op"><</span> <span class="num">101</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t320" class="run"><span class="n"><a href="#t320">320</a></span><span class="t"> <span class="nam">b</span> <span class="op">=</span> <span class="nam">age_by_brackets_dic_18</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t321" class="pln"><span class="n"><a href="#t321">321</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t322" class="run"><span class="n"><a href="#t322">322</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">n</span> <span class="op">*</span> <span class="nam">age_distr_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">/</span><span class="nam">len</span><span class="op">(</span><span class="nam">age_brackets_18</span><span class="op">[</span><span class="nam">b</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t323" class="run"><span class="n"><a href="#t323">323</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">*</span> <span class="nam">est_ltcf_user_by_age_brackets_perc</span><span class="op">[</span><span class="str">'85-100'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t324" class="run"><span class="n"><a href="#t324">324</a></span><span class="t"> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">math</span><span class="op">.</span><span class="nam">ceil</span><span class="op">(</span><span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t325" class="pln"><span class="n"><a href="#t325">325</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t326" class="run"><span class="n"><a href="#t326">326</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t327" class="mis show_mis"><span class="n"><a href="#t327">327</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">np</span><span class="op">.</span><span class="nam">sum</span><span class="op">(</span><span class="op">[</span><span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">expected_users_by_age</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t328" class="pln"><span class="n"><a href="#t328">328</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t329" class="run"><span class="n"><a href="#t329">329</a></span><span class="t"> <span class="nam">KC_resident_size_distr</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_usa_long_term_care_facility_residents_distr</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t330" class="run"><span class="n"><a href="#t330">330</a></span><span class="t"> <span class="nam">KC_resident_size_distr</span> <span class="op">=</span> <span class="nam">spb</span><span class="op">.</span><span class="nam">norm_dic</span><span class="op">(</span><span class="nam">KC_resident_size_distr</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t331" class="run"><span class="n"><a href="#t331">331</a></span><span class="t"> <span class="nam">KC_residents_size_brackets</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_usa_long_term_care_facility_residents_distr_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t332" class="pln"><span class="n"><a href="#t332">332</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t333" class="run"><span class="n"><a href="#t333">333</a></span><span class="t"> <span class="nam">all_residents</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t334" class="run"><span class="n"><a href="#t334">334</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">expected_users_by_age</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t335" class="run"><span class="n"><a href="#t335">335</a></span><span class="t"> <span class="nam">all_residents</span> <span class="op">+=</span> <span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">*</span> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t336" class="run"><span class="n"><a href="#t336">336</a></span><span class="t"> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">shuffle</span><span class="op">(</span><span class="nam">all_residents</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t337" class="pln"><span class="n"><a href="#t337">337</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t338" class="pln"><span class="n"><a href="#t338">338</a></span><span class="t"> <span class="com"># place residents in facilities</span> </span><span class="r"></span></p>
<p id="t339" class="run"><span class="n"><a href="#t339">339</a></span><span class="t"> <span class="nam">facilities</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t340" class="pln"><span class="n"><a href="#t340">340</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t341" class="run"><span class="n"><a href="#t341">341</a></span><span class="t"> <span class="nam">size_bracket_keys</span> <span class="op">=</span> <span class="nam">sorted</span><span class="op">(</span><span class="op">[</span><span class="nam">k</span> <span class="key">for</span> <span class="nam">k</span> <span class="key">in</span> <span class="nam">KC_resident_size_distr</span><span class="op">.</span><span class="nam">keys</span><span class="op">(</span><span class="op">)</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t342" class="run"><span class="n"><a href="#t342">342</a></span><span class="t"> <span class="nam">size_distr_array</span> <span class="op">=</span> <span class="op">[</span><span class="nam">KC_resident_size_distr</span><span class="op">[</span><span class="nam">k</span><span class="op">]</span> <span class="key">for</span> <span class="nam">k</span> <span class="key">in</span> <span class="nam">size_bracket_keys</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t343" class="run"><span class="n"><a href="#t343">343</a></span><span class="t"> <span class="key">while</span> <span class="nam">len</span><span class="op">(</span><span class="nam">all_residents</span><span class="op">)</span> <span class="op">></span> <span class="num">0</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t344" class="pln"><span class="n"><a href="#t344">344</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t345" class="run"><span class="n"><a href="#t345">345</a></span><span class="t"> <span class="nam">sb</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">choice</span><span class="op">(</span><span class="nam">size_bracket_keys</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="nam">size_distr_array</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t346" class="run"><span class="n"><a href="#t346">346</a></span><span class="t"> <span class="nam">sb_range</span> <span class="op">=</span> <span class="nam">KC_residents_size_brackets</span><span class="op">[</span><span class="nam">sb</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t347" class="run"><span class="n"><a href="#t347">347</a></span><span class="t"> <span class="nam">size</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">choice</span><span class="op">(</span><span class="nam">sb_range</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t348" class="pln"><span class="n"><a href="#t348">348</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t349" class="pln"><span class="n"><a href="#t349">349</a></span><span class="t"> <span class="com"># size = int(np.random.choice(KC_ltcf_sizes))</span> </span><span class="r"></span></p>
<p id="t350" class="run"><span class="n"><a href="#t350">350</a></span><span class="t"> <span class="key">if</span> <span class="nam">size</span> <span class="op">></span> <span class="nam">len</span><span class="op">(</span><span class="nam">all_residents</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t351" class="run"><span class="n"><a href="#t351">351</a></span><span class="t"> <span class="nam">size</span> <span class="op">=</span> <span class="nam">len</span><span class="op">(</span><span class="nam">all_residents</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t352" class="pln"><span class="n"><a href="#t352">352</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t353" class="run"><span class="n"><a href="#t353">353</a></span><span class="t"> <span class="nam">new_facility</span> <span class="op">=</span> <span class="nam">all_residents</span><span class="op">[</span><span class="num">0</span><span class="op">:</span><span class="nam">size</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t354" class="run"><span class="n"><a href="#t354">354</a></span><span class="t"> <span class="nam">facilities</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">new_facility</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t355" class="run"><span class="n"><a href="#t355">355</a></span><span class="t"> <span class="nam">all_residents</span> <span class="op">=</span> <span class="nam">all_residents</span><span class="op">[</span><span class="nam">size</span><span class="op">:</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t356" class="pln"><span class="n"><a href="#t356">356</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t357" class="run"><span class="n"><a href="#t357">357</a></span><span class="t"> <span class="nam">max_age</span> <span class="op">=</span> <span class="num">100</span> </span><span class="r"></span></p>
<p id="t358" class="pln"><span class="n"><a href="#t358">358</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t359" class="run"><span class="n"><a href="#t359">359</a></span><span class="t"> <span class="nam">expected_age_distr</span> <span class="op">=</span> <span class="nam">dict</span><span class="op">.</span><span class="nam">fromkeys</span><span class="op">(</span><span class="nam">np</span><span class="op">.</span><span class="nam">arange</span><span class="op">(</span><span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span><span class="op">,</span> <span class="num">0</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t360" class="run"><span class="n"><a href="#t360">360</a></span><span class="t"> <span class="nam">expected_age_count</span> <span class="op">=</span> <span class="nam">dict</span><span class="op">.</span><span class="nam">fromkeys</span><span class="op">(</span><span class="nam">np</span><span class="op">.</span><span class="nam">arange</span><span class="op">(</span><span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span><span class="op">,</span> <span class="num">0</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t361" class="pln"><span class="n"><a href="#t361">361</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t362" class="pln"><span class="n"><a href="#t362">362</a></span><span class="t"> <span class="com"># adjust age distribution for those already created</span> </span><span class="r"></span></p>
<p id="t363" class="run"><span class="n"><a href="#t363">363</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">expected_age_distr</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t364" class="run"><span class="n"><a href="#t364">364</a></span><span class="t"> <span class="nam">expected_age_distr</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">age_distr_16</span><span class="op">[</span><span class="nam">age_by_brackets_dic_16</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">]</span><span class="op">/</span><span class="nam">len</span><span class="op">(</span><span class="nam">age_brackets_16</span><span class="op">[</span><span class="nam">age_by_brackets_dic_16</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t365" class="run"><span class="n"><a href="#t365">365</a></span><span class="t"> <span class="nam">expected_age_count</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">n</span> <span class="op">*</span> <span class="nam">expected_age_distr</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t366" class="pln"><span class="n"><a href="#t366">366</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t367" class="run"><span class="n"><a href="#t367">367</a></span><span class="t"> <span class="nam">ltcf_adjusted_age_count</span> <span class="op">=</span> <span class="nam">deepcopy</span><span class="op">(</span><span class="nam">expected_age_count</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t368" class="run"><span class="n"><a href="#t368">368</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">expected_users_by_age</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t369" class="run"><span class="n"><a href="#t369">369</a></span><span class="t"> <span class="nam">ltcf_adjusted_age_count</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">-=</span> <span class="nam">expected_users_by_age</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t370" class="run"><span class="n"><a href="#t370">370</a></span><span class="t"> <span class="nam">ltcf_adjusted_age_distr_dict</span> <span class="op">=</span> <span class="nam">spb</span><span class="op">.</span><span class="nam">norm_dic</span><span class="op">(</span><span class="nam">ltcf_adjusted_age_count</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t371" class="run"><span class="n"><a href="#t371">371</a></span><span class="t"> <span class="nam">ltcf_adjusted_age_distr</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">array</span><span class="op">(</span><span class="op">[</span><span class="nam">ltcf_adjusted_age_distr_dict</span><span class="op">[</span><span class="nam">i</span><span class="op">]</span> <span class="key">for</span> <span class="nam">i</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t372" class="pln"><span class="n"><a href="#t372">372</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t373" class="run"><span class="n"><a href="#t373">373</a></span><span class="t"> <span class="nam">exp_age_distr</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">array</span><span class="op">(</span><span class="op">[</span><span class="nam">expected_age_distr</span><span class="op">[</span><span class="nam">i</span><span class="op">]</span> <span class="key">for</span> <span class="nam">i</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span><span class="op">]</span><span class="op">,</span> <span class="nam">dtype</span><span class="op">=</span><span class="nam">np</span><span class="op">.</span><span class="nam">float64</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t374" class="pln"><span class="n"><a href="#t374">374</a></span><span class="t"> <span class="com"># exp_age_distr = np.array(list(expected_age_distr.values()), dtype=np.float64)</span> </span><span class="r"></span></p>
<p id="t375" class="pln"><span class="n"><a href="#t375">375</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t376" class="pln"><span class="n"><a href="#t376">376</a></span><span class="t"> <span class="com"># build rest of the population</span> </span><span class="r"></span></p>
<p id="t377" class="run"><span class="n"><a href="#t377">377</a></span><span class="t"> <span class="nam">n_nonltcf</span> <span class="op">=</span> <span class="nam">n</span> <span class="op">-</span> <span class="nam">np</span><span class="op">.</span><span class="nam">sum</span><span class="op">(</span><span class="op">[</span><span class="nam">len</span><span class="op">(</span><span class="nam">f</span><span class="op">)</span> <span class="key">for</span> <span class="nam">f</span> <span class="key">in</span> <span class="nam">facilities</span><span class="op">]</span><span class="op">)</span> <span class="com"># remove those placed in care homes</span> </span><span class="r"></span></p>
<p id="t378" class="pln"><span class="n"><a href="#t378">378</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t379" class="run"><span class="n"><a href="#t379">379</a></span><span class="t"> <span class="nam">household_size_distr</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_household_size_distr</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t380" class="run"><span class="n"><a href="#t380">380</a></span><span class="t"> <span class="nam">hh_sizes</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">generate_household_sizes_from_fixed_pop_size</span><span class="op">(</span><span class="nam">n_nonltcf</span><span class="op">,</span> <span class="nam">household_size_distr</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t381" class="run"><span class="n"><a href="#t381">381</a></span><span class="t"> <span class="nam">hha_brackets</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_head_age_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t382" class="run"><span class="n"><a href="#t382">382</a></span><span class="t"> <span class="nam">hha_by_size</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_head_age_by_size_distr</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t383" class="pln"><span class="n"><a href="#t383">383</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t384" class="run"><span class="n"><a href="#t384">384</a></span><span class="t"> <span class="nam">contact_matrix_dic</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_contact_matrix_dic</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">sheet_name</span><span class="op">=</span><span class="nam">sheet_name</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t385" class="pln"><span class="n"><a href="#t385">385</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t386" class="run"><span class="n"><a href="#t386">386</a></span><span class="t"> <span class="nam">homes_dic</span><span class="op">,</span> <span class="nam">homes</span> <span class="op">=</span> <span class="nam">custom_generate_all_households</span><span class="op">(</span><span class="nam">n_nonltcf</span><span class="op">,</span> <span class="nam">hh_sizes</span><span class="op">,</span> <span class="nam">hha_by_size</span><span class="op">,</span> <span class="nam">hha_brackets</span><span class="op">,</span> <span class="nam">age_brackets_16</span><span class="op">,</span> <span class="nam">age_by_brackets_dic_16</span><span class="op">,</span> <span class="nam">contact_matrix_dic</span><span class="op">,</span> <span class="nam">ltcf_adjusted_age_distr</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t387" class="run"><span class="n"><a href="#t387">387</a></span><span class="t"> <span class="nam">homes</span> <span class="op">=</span> <span class="nam">facilities</span> <span class="op">+</span> <span class="nam">homes</span> </span><span class="r"></span></p>
<p id="t388" class="pln"><span class="n"><a href="#t388">388</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t389" class="run"><span class="n"><a href="#t389">389</a></span><span class="t"> <span class="nam">homes_by_uids</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">assign_uids_by_homes</span><span class="op">(</span><span class="nam">homes</span><span class="op">)</span> <span class="com">#include facilities to assign ids</span> </span><span class="r"></span></p>
<p id="t390" class="run"><span class="n"><a href="#t390">390</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'homes by uids'</span><span class="op">,</span> <span class="nam">homes_by_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t391" class="run"><span class="n"><a href="#t391">391</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'facilities'</span><span class="op">,</span> <span class="nam">facilities</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t392" class="run"><span class="n"><a href="#t392">392</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'age_by_uid_dic'</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t393" class="run"><span class="n"><a href="#t393">393</a></span><span class="t"> <span class="nam">new_ages_count</span> <span class="op">=</span> <span class="nam">Counter</span><span class="op">(</span><span class="nam">age_by_uid_dic</span><span class="op">.</span><span class="nam">values</span><span class="op">(</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t394" class="pln"><span class="n"><a href="#t394">394</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t395" class="run"><span class="n"><a href="#t395">395</a></span><span class="t"> <span class="nam">facilities_by_uids</span> <span class="op">=</span> <span class="nam">homes_by_uids</span><span class="op">[</span><span class="num">0</span><span class="op">:</span><span class="nam">len</span><span class="op">(</span><span class="nam">facilities</span><span class="op">)</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t396" class="pln"><span class="n"><a href="#t396">396</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t397" class="run"><span class="n"><a href="#t397">397</a></span><span class="t"> <span class="key">if</span> <span class="nam">plot</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t398" class="pln"><span class="n"><a href="#t398">398</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t399" class="mis show_mis"><span class="n"><a href="#t399">399</a></span><span class="t"> <span class="nam">fig</span> <span class="op">=</span> <span class="nam">plt</span><span class="op">.</span><span class="nam">figure</span><span class="op">(</span><span class="nam">figsize</span><span class="op">=</span><span class="op">(</span><span class="num">7</span><span class="op">,</span> <span class="num">5</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t400" class="mis show_mis"><span class="n"><a href="#t400">400</a></span><span class="t"> <span class="nam">ax</span> <span class="op">=</span> <span class="nam">fig</span><span class="op">.</span><span class="nam">add_subplot</span><span class="op">(</span><span class="num">111</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t401" class="mis show_mis"><span class="n"><a href="#t401">401</a></span><span class="t"> <span class="nam">x</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">arange</span><span class="op">(</span><span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t402" class="mis show_mis"><span class="n"><a href="#t402">402</a></span><span class="t"> <span class="nam">y_exp</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">zeros</span><span class="op">(</span><span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t403" class="mis show_mis"><span class="n"><a href="#t403">403</a></span><span class="t"> <span class="nam">y_sim</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">zeros</span><span class="op">(</span><span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t404" class="mis show_mis"><span class="n"><a href="#t404">404</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">x</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t405" class="mis show_mis"><span class="n"><a href="#t405">405</a></span><span class="t"> <span class="nam">y_exp</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">expected_age_distr</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t406" class="mis show_mis"><span class="n"><a href="#t406">406</a></span><span class="t"> <span class="nam">y_sim</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="op">=</span> <span class="nam">new_ages_count</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span><span class="op">/</span><span class="nam">n</span> </span><span class="r"></span></p>
<p id="t407" class="mis show_mis"><span class="n"><a href="#t407">407</a></span><span class="t"> <span class="nam">ax</span><span class="op">.</span><span class="nam">plot</span><span class="op">(</span><span class="nam">x</span><span class="op">,</span> <span class="nam">y_exp</span><span class="op">,</span> <span class="nam">color</span><span class="op">=</span><span class="str">'k'</span><span class="op">,</span> <span class="nam">label</span><span class="op">=</span><span class="str">'Expected'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t408" class="mis show_mis"><span class="n"><a href="#t408">408</a></span><span class="t"> <span class="nam">ax</span><span class="op">.</span><span class="nam">plot</span><span class="op">(</span><span class="nam">x</span><span class="op">,</span> <span class="nam">y_sim</span><span class="op">,</span> <span class="nam">color</span><span class="op">=</span><span class="str">'teal'</span><span class="op">,</span> <span class="nam">label</span><span class="op">=</span><span class="str">'Simulated'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t409" class="mis show_mis"><span class="n"><a href="#t409">409</a></span><span class="t"> <span class="nam">leg</span> <span class="op">=</span> <span class="nam">ax</span><span class="op">.</span><span class="nam">legend</span><span class="op">(</span><span class="nam">fontsize</span><span class="op">=</span><span class="num">18</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t410" class="mis show_mis"><span class="n"><a href="#t410">410</a></span><span class="t"> <span class="nam">leg</span><span class="op">.</span><span class="nam">draw_frame</span><span class="op">(</span><span class="key">False</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t411" class="mis show_mis"><span class="n"><a href="#t411">411</a></span><span class="t"> <span class="nam">ax</span><span class="op">.</span><span class="nam">set_xlim</span><span class="op">(</span><span class="num">0</span><span class="op">,</span> <span class="nam">max_age</span><span class="op">+</span><span class="num">1</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t412" class="mis show_mis"><span class="n"><a href="#t412">412</a></span><span class="t"> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="num">6</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t413" class="mis show_mis"><span class="n"><a href="#t413">413</a></span><span class="t"> <span class="nam">ax</span><span class="op">.</span><span class="nam">axvline</span><span class="op">(</span><span class="nam">x</span><span class="op">=</span><span class="nam">a</span><span class="op">,</span> <span class="nam">ymin</span><span class="op">=</span><span class="num">0</span><span class="op">,</span> <span class="nam">ymax</span><span class="op">=</span><span class="num">1</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t414" class="mis show_mis"><span class="n"><a href="#t414">414</a></span><span class="t"> <span class="nam">plt</span><span class="op">.</span><span class="nam">show</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t415" class="pln"><span class="n"><a href="#t415">415</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t416" class="pln"><span class="n"><a href="#t416">416</a></span><span class="t"> <span class="com"># Make a dictionary listing out uids of people by their age</span> </span><span class="r"></span></p>
<p id="t417" class="run"><span class="n"><a href="#t417">417</a></span><span class="t"> <span class="nam">uids_by_age_dic</span> <span class="op">=</span> <span class="nam">spb</span><span class="op">.</span><span class="nam">get_ids_by_age_dic</span><span class="op">(</span><span class="nam">age_by_uid_dic</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t418" class="pln"><span class="n"><a href="#t418">418</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t419" class="pln"><span class="n"><a href="#t419">419</a></span><span class="t"> <span class="com"># Generate school sizes</span> </span><span class="r"></span></p>
<p id="t420" class="run"><span class="n"><a href="#t420">420</a></span><span class="t"> <span class="nam">school_sizes_count_by_brackets</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_school_size_distr_by_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">counts_available</span><span class="op">=</span><span class="nam">school_enrollment_counts_available</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t421" class="run"><span class="n"><a href="#t421">421</a></span><span class="t"> <span class="nam">school_size_brackets</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_school_size_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t422" class="pln"><span class="n"><a href="#t422">422</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t423" class="pln"><span class="n"><a href="#t423">423</a></span><span class="t"> <span class="com"># Figure out who's going to school as a student with enrollment rates (gets called inside sp.get_uids_in_school)</span> </span><span class="r"></span></p>
<p id="t424" class="run"><span class="n"><a href="#t424">424</a></span><span class="t"> <span class="nam">uids_in_school</span><span class="op">,</span> <span class="nam">uids_in_school_by_age</span><span class="op">,</span> <span class="nam">ages_in_school_count</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">get_uids_in_school</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">n_nonltcf</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">homes_by_uids</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> <span class="com"># this will call in school enrollment rates</span> </span><span class="r"></span></p>
<p id="t425" class="pln"><span class="n"><a href="#t425">425</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t426" class="pln"><span class="n"><a href="#t426">426</a></span><span class="t"> <span class="com"># Get school sizes</span> </span><span class="r"></span></p>
<p id="t427" class="run"><span class="n"><a href="#t427">427</a></span><span class="t"> <span class="nam">gen_school_sizes</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">generate_school_sizes</span><span class="op">(</span><span class="nam">school_sizes_count_by_brackets</span><span class="op">,</span> <span class="nam">school_size_brackets</span><span class="op">,</span> <span class="nam">uids_in_school</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t428" class="pln"><span class="n"><a href="#t428">428</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t429" class="pln"><span class="n"><a href="#t429">429</a></span><span class="t"> <span class="com"># Assign students to school</span> </span><span class="r"></span></p>
<p id="t430" class="run"><span class="n"><a href="#t430">430</a></span><span class="t"> <span class="nam">gen_schools</span><span class="op">,</span> <span class="nam">gen_school_uids</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">send_students_to_school</span><span class="op">(</span><span class="nam">gen_school_sizes</span><span class="op">,</span> <span class="nam">uids_in_school</span><span class="op">,</span> <span class="nam">uids_in_school_by_age</span><span class="op">,</span> <span class="nam">ages_in_school_count</span><span class="op">,</span> <span class="nam">age_brackets_16</span><span class="op">,</span> <span class="nam">age_by_brackets_dic_16</span><span class="op">,</span> <span class="nam">contact_matrix_dic</span><span class="op">,</span> <span class="nam">verbose</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t431" class="pln"><span class="n"><a href="#t431">431</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t432" class="pln"><span class="n"><a href="#t432">432</a></span><span class="t"> <span class="com"># Get employment rates</span> </span><span class="r"></span></p>
<p id="t433" class="run"><span class="n"><a href="#t433">433</a></span><span class="t"> <span class="nam">employment_rates</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_employment_rates</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t434" class="pln"><span class="n"><a href="#t434">434</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t435" class="pln"><span class="n"><a href="#t435">435</a></span><span class="t"> <span class="com"># Find people who can be workers (removing everyone who is currently a student)</span> </span><span class="r"></span></p>
<p id="t436" class="run"><span class="n"><a href="#t436">436</a></span><span class="t"> <span class="nam">potential_worker_uids</span><span class="op">,</span> <span class="nam">potential_worker_uids_by_age</span><span class="op">,</span> <span class="nam">potential_worker_ages_left_count</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">get_uids_potential_workers</span><span class="op">(</span><span class="nam">gen_school_uids</span><span class="op">,</span> <span class="nam">employment_rates</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t437" class="run"><span class="n"><a href="#t437">437</a></span><span class="t"> <span class="nam">workers_by_age_to_assign_count</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">get_workers_by_age_to_assign</span><span class="op">(</span><span class="nam">employment_rates</span><span class="op">,</span> <span class="nam">potential_worker_ages_left_count</span><span class="op">,</span> <span class="nam">uids_by_age_dic</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t438" class="pln"><span class="n"><a href="#t438">438</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t439" class="pln"><span class="n"><a href="#t439">439</a></span><span class="t"> <span class="com"># Removing facilities residents from potential workers</span> </span><span class="r"></span></p>
<p id="t440" class="run"><span class="n"><a href="#t440">440</a></span><span class="t"> <span class="key">for</span> <span class="nam">nf</span><span class="op">,</span> <span class="nam">fc</span> <span class="key">in</span> <span class="nam">enumerate</span><span class="op">(</span><span class="nam">facilities_by_uids</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t441" class="run"><span class="n"><a href="#t441">441</a></span><span class="t"> <span class="key">for</span> <span class="nam">uid</span> <span class="key">in</span> <span class="nam">fc</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t442" class="run"><span class="n"><a href="#t442">442</a></span><span class="t"> <span class="nam">aindex</span> <span class="op">=</span> <span class="nam">age_by_uid_dic</span><span class="op">[</span><span class="nam">uid</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t443" class="run"><span class="n"><a href="#t443">443</a></span><span class="t"> <span class="key">if</span> <span class="nam">uid</span> <span class="key">in</span> <span class="nam">potential_worker_uids</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t444" class="run"><span class="n"><a href="#t444">444</a></span><span class="t"> <span class="nam">potential_worker_uids_by_age</span><span class="op">[</span><span class="nam">aindex</span><span class="op">]</span><span class="op">.</span><span class="nam">remove</span><span class="op">(</span><span class="nam">uid</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t445" class="run"><span class="n"><a href="#t445">445</a></span><span class="t"> <span class="nam">potential_worker_uids</span><span class="op">.</span><span class="nam">pop</span><span class="op">(</span><span class="nam">uid</span><span class="op">,</span> <span class="key">None</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t446" class="run"><span class="n"><a href="#t446">446</a></span><span class="t"> <span class="key">if</span> <span class="nam">workers_by_age_to_assign_count</span><span class="op">[</span><span class="nam">aindex</span><span class="op">]</span> <span class="op">></span> <span class="num">0</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t447" class="run"><span class="n"><a href="#t447">447</a></span><span class="t"> <span class="nam">workers_by_age_to_assign_count</span><span class="op">[</span><span class="nam">aindex</span><span class="op">]</span> <span class="op">-=</span> <span class="num">1</span> </span><span class="r"></span></p>
<p id="t448" class="pln"><span class="n"><a href="#t448">448</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t449" class="pln"><span class="n"><a href="#t449">449</a></span><span class="t"> <span class="com"># Assign teachers and update school lists</span> </span><span class="r"></span></p>
<p id="t450" class="pln"><span class="n"><a href="#t450">450</a></span><span class="t"> <span class="com"># gen_schools, gen_school_uids, potential_worker_uids, potential_worker_uids_by_age, workers_by_age_to_assign_count = spcnx.assign_teachers_to_work(gen_schools, gen_school_uids, employment_rates, workers_by_age_to_assign_count, potential_worker_uids, potential_worker_uids_by_age, potential_worker_ages_left_count, verbose=verbose)</span> </span><span class="r"></span></p>
<p id="t451" class="run"><span class="n"><a href="#t451">451</a></span><span class="t"> <span class="nam">gen_teachers</span><span class="op">,</span> <span class="nam">gen_teacher_uids</span><span class="op">,</span> <span class="nam">potential_worker_uids</span><span class="op">,</span> <span class="nam">potential_worker_uids_by_age</span><span class="op">,</span> <span class="nam">workers_by_age_to_assign_count</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">assign_teachers_to_work</span><span class="op">(</span><span class="nam">gen_schools</span><span class="op">,</span> <span class="nam">gen_school_uids</span><span class="op">,</span> <span class="nam">employment_rates</span><span class="op">,</span> <span class="nam">workers_by_age_to_assign_count</span><span class="op">,</span> <span class="nam">potential_worker_uids</span><span class="op">,</span> <span class="nam">potential_worker_uids_by_age</span><span class="op">,</span> <span class="nam">potential_worker_ages_left_count</span><span class="op">,</span> <span class="nam">verbose</span><span class="op">=</span><span class="nam">verbose</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t452" class="pln"><span class="n"><a href="#t452">452</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t453" class="pln"><span class="n"><a href="#t453">453</a></span><span class="t"> <span class="com"># Assign facilities care staff from 20 to 59</span> </span><span class="r"></span></p>
<p id="t454" class="pln"><span class="n"><a href="#t454">454</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t455" class="run"><span class="n"><a href="#t455">455</a></span><span class="t"> <span class="nam">KC_ratio_distr</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_usa_long_term_care_facility_resident_to_staff_ratios_distr</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t456" class="run"><span class="n"><a href="#t456">456</a></span><span class="t"> <span class="nam">KC_ratio_distr</span> <span class="op">=</span> <span class="nam">spb</span><span class="op">.</span><span class="nam">norm_dic</span><span class="op">(</span><span class="nam">KC_ratio_distr</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t457" class="run"><span class="n"><a href="#t457">457</a></span><span class="t"> <span class="nam">KC_ratio_brackets</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_usa_long_term_care_facility_resident_to_staff_ratios_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">=</span><span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t458" class="pln"><span class="n"><a href="#t458">458</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t459" class="run"><span class="n"><a href="#t459">459</a></span><span class="t"> <span class="nam">facilities_staff</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t460" class="run"><span class="n"><a href="#t460">460</a></span><span class="t"> <span class="nam">facilities_staff_uids</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t461" class="pln"><span class="n"><a href="#t461">461</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t462" class="run"><span class="n"><a href="#t462">462</a></span><span class="t"> <span class="nam">sorted_ratio_keys</span> <span class="op">=</span> <span class="nam">sorted</span><span class="op">(</span><span class="op">[</span><span class="nam">k</span> <span class="key">for</span> <span class="nam">k</span> <span class="key">in</span> <span class="nam">KC_ratio_distr</span><span class="op">.</span><span class="nam">keys</span><span class="op">(</span><span class="op">)</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t463" class="run"><span class="n"><a href="#t463">463</a></span><span class="t"> <span class="nam">sorted_ratio_array</span> <span class="op">=</span> <span class="op">[</span><span class="nam">KC_ratio_distr</span><span class="op">[</span><span class="nam">k</span><span class="op">]</span> <span class="key">for</span> <span class="nam">k</span> <span class="key">in</span> <span class="nam">sorted_ratio_keys</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t464" class="run"><span class="n"><a href="#t464">464</a></span><span class="t"> <span class="nam">staff_age_range</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">arange</span><span class="op">(</span><span class="num">20</span><span class="op">,</span> <span class="num">60</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t465" class="run"><span class="n"><a href="#t465">465</a></span><span class="t"> <span class="key">for</span> <span class="nam">nf</span><span class="op">,</span> <span class="nam">fc</span> <span class="key">in</span> <span class="nam">enumerate</span><span class="op">(</span><span class="nam">facilities</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t466" class="run"><span class="n"><a href="#t466">466</a></span><span class="t"> <span class="nam">n_residents</span> <span class="op">=</span> <span class="nam">len</span><span class="op">(</span><span class="nam">fc</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t467" class="pln"><span class="n"><a href="#t467">467</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t468" class="run"><span class="n"><a href="#t468">468</a></span><span class="t"> <span class="nam">sb</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">choice</span><span class="op">(</span><span class="nam">sorted_ratio_keys</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="nam">sorted_ratio_array</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t469" class="run"><span class="n"><a href="#t469">469</a></span><span class="t"> <span class="nam">sb_range</span> <span class="op">=</span> <span class="nam">KC_ratio_brackets</span><span class="op">[</span><span class="nam">sb</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t470" class="run"><span class="n"><a href="#t470">470</a></span><span class="t"> <span class="nam">resident_staff_ratio</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">mean</span><span class="op">(</span><span class="nam">sb_range</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t471" class="pln"><span class="n"><a href="#t471">471</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t472" class="pln"><span class="n"><a href="#t472">472</a></span><span class="t"> <span class="com"># if using raw staff totals in residents to staff ratios divide rato by 3 to split staff into 3 8 hour shifts at minimum</span> </span><span class="r"></span></p>
<p id="t473" class="run"><span class="n"><a href="#t473">473</a></span><span class="t"> <span class="nam">resident_staff_ratio</span> <span class="op">=</span> <span class="nam">resident_staff_ratio</span><span class="op">/</span><span class="num">3.</span> </span><span class="r"></span></p>
<p id="t474" class="pln"><span class="n"><a href="#t474">474</a></span><span class="t"> <span class="com"># resident_staff_ratio = np.random.choice(KC_resident_staff_ratios)</span> </span><span class="r"></span></p>
<p id="t475" class="pln"><span class="n"><a href="#t475">475</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t476" class="run"><span class="n"><a href="#t476">476</a></span><span class="t"> <span class="nam">n_staff</span> <span class="op">=</span> <span class="nam">int</span><span class="op">(</span><span class="nam">math</span><span class="op">.</span><span class="nam">ceil</span><span class="op">(</span><span class="nam">n_residents</span><span class="op">/</span><span class="nam">resident_staff_ratio</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t477" class="run"><span class="n"><a href="#t477">477</a></span><span class="t"> <span class="nam">new_staff</span><span class="op">,</span> <span class="nam">new_staff_uids</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span><span class="op">,</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t478" class="pln"><span class="n"><a href="#t478">478</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t479" class="run"><span class="n"><a href="#t479">479</a></span><span class="t"> <span class="key">for</span> <span class="nam">i</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="nam">n_staff</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t480" class="run"><span class="n"><a href="#t480">480</a></span><span class="t"> <span class="nam">a_prob</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">array</span><span class="op">(</span><span class="op">[</span><span class="nam">workers_by_age_to_assign_count</span><span class="op">[</span><span class="nam">a</span><span class="op">]</span> <span class="key">for</span> <span class="nam">a</span> <span class="key">in</span> <span class="nam">staff_age_range</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t481" class="run"><span class="n"><a href="#t481">481</a></span><span class="t"> <span class="nam">a_prob</span> <span class="op">=</span> <span class="nam">a_prob</span><span class="op">/</span><span class="nam">np</span><span class="op">.</span><span class="nam">sum</span><span class="op">(</span><span class="nam">a_prob</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t482" class="run"><span class="n"><a href="#t482">482</a></span><span class="t"> <span class="nam">aindex</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">choice</span><span class="op">(</span><span class="nam">a</span><span class="op">=</span><span class="nam">staff_age_range</span><span class="op">,</span> <span class="nam">p</span><span class="op">=</span><span class="nam">a_prob</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t483" class="pln"><span class="n"><a href="#t483">483</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t484" class="run"><span class="n"><a href="#t484">484</a></span><span class="t"> <span class="nam">uid</span> <span class="op">=</span> <span class="nam">potential_worker_uids_by_age</span><span class="op">[</span><span class="nam">aindex</span><span class="op">]</span><span class="op">[</span><span class="num">0</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t485" class="run"><span class="n"><a href="#t485">485</a></span><span class="t"> <span class="nam">potential_worker_uids_by_age</span><span class="op">[</span><span class="nam">aindex</span><span class="op">]</span><span class="op">.</span><span class="nam">remove</span><span class="op">(</span><span class="nam">uid</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t486" class="run"><span class="n"><a href="#t486">486</a></span><span class="t"> <span class="nam">potential_worker_uids</span><span class="op">.</span><span class="nam">pop</span><span class="op">(</span><span class="nam">uid</span><span class="op">,</span> <span class="key">None</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t487" class="run"><span class="n"><a href="#t487">487</a></span><span class="t"> <span class="nam">workers_by_age_to_assign_count</span><span class="op">[</span><span class="nam">aindex</span><span class="op">]</span> <span class="op">-=</span> <span class="num">1</span> </span><span class="r"></span></p>
<p id="t488" class="pln"><span class="n"><a href="#t488">488</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t489" class="run"><span class="n"><a href="#t489">489</a></span><span class="t"> <span class="nam">new_staff</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">aindex</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t490" class="run"><span class="n"><a href="#t490">490</a></span><span class="t"> <span class="nam">new_staff_uids</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">uid</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t491" class="pln"><span class="n"><a href="#t491">491</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t492" class="run"><span class="n"><a href="#t492">492</a></span><span class="t"> <span class="nam">facilities_staff</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">new_staff</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t493" class="run"><span class="n"><a href="#t493">493</a></span><span class="t"> <span class="nam">facilities_staff_uids</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">new_staff_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t494" class="pln"><span class="n"><a href="#t494">494</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t495" class="run"><span class="n"><a href="#t495">495</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t496" class="mis show_mis"><span class="n"><a href="#t496">496</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">len</span><span class="op">(</span><span class="nam">facilities_staff_uids</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t497" class="mis show_mis"><span class="n"><a href="#t497">497</a></span><span class="t"> <span class="key">for</span> <span class="nam">nf</span><span class="op">,</span> <span class="nam">fc</span> <span class="key">in</span> <span class="nam">enumerate</span><span class="op">(</span><span class="nam">facilities</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t498" class="mis show_mis"><span class="n"><a href="#t498">498</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">fc</span><span class="op">,</span> <span class="nam">facilities_staff</span><span class="op">[</span><span class="nam">nf</span><span class="op">]</span><span class="op">,</span> <span class="nam">len</span><span class="op">(</span><span class="nam">fc</span><span class="op">)</span><span class="op">/</span><span class="nam">len</span><span class="op">(</span><span class="nam">facilities_staff</span><span class="op">[</span><span class="nam">nf</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t499" class="pln"><span class="n"><a href="#t499">499</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t500" class="pln"><span class="n"><a href="#t500">500</a></span><span class="t"> <span class="com"># Generate non-school workplace sizes needed to send everyone to work</span> </span><span class="r"></span></p>
<p id="t501" class="run"><span class="n"><a href="#t501">501</a></span><span class="t"> <span class="nam">workplace_size_brackets</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_workplace_size_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t502" class="run"><span class="n"><a href="#t502">502</a></span><span class="t"> <span class="nam">workplace_size_distr_by_brackets</span> <span class="op">=</span> <span class="nam">spdata</span><span class="op">.</span><span class="nam">get_workplace_size_distr_by_brackets</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">=</span><span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">=</span><span class="nam">country_location</span><span class="op">,</span> <span class="nam">use_default</span><span class="op">=</span><span class="nam">use_default</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t503" class="run"><span class="n"><a href="#t503">503</a></span><span class="t"> <span class="nam">workplace_sizes</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">generate_workplace_sizes</span><span class="op">(</span><span class="nam">workplace_size_distr_by_brackets</span><span class="op">,</span> <span class="nam">workplace_size_brackets</span><span class="op">,</span> <span class="nam">workers_by_age_to_assign_count</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t504" class="pln"><span class="n"><a href="#t504">504</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t505" class="pln"><span class="n"><a href="#t505">505</a></span><span class="t"> <span class="com"># Assign all workers who are not staff at schools to workplaces</span> </span><span class="r"></span></p>
<p id="t506" class="run"><span class="n"><a href="#t506">506</a></span><span class="t"> <span class="nam">gen_workplaces</span><span class="op">,</span> <span class="nam">gen_workplace_uids</span><span class="op">,</span> <span class="nam">potential_worker_uids</span><span class="op">,</span> <span class="nam">potential_worker_uids_by_age</span><span class="op">,</span> <span class="nam">workers_by_age_to_assign_count</span> <span class="op">=</span> <span class="nam">spcnx</span><span class="op">.</span><span class="nam">assign_rest_of_workers</span><span class="op">(</span><span class="nam">workplace_sizes</span><span class="op">,</span> <span class="nam">potential_worker_uids</span><span class="op">,</span> <span class="nam">potential_worker_uids_by_age</span><span class="op">,</span> <span class="nam">workers_by_age_to_assign_count</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">age_brackets_16</span><span class="op">,</span> <span class="nam">age_by_brackets_dic_16</span><span class="op">,</span> <span class="nam">contact_matrix_dic</span><span class="op">,</span> <span class="nam">verbose</span><span class="op">=</span><span class="nam">verbose</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t507" class="pln"><span class="n"><a href="#t507">507</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t508" class="pln"><span class="n"><a href="#t508">508</a></span><span class="t"> <span class="com"># remove facilities from homes to write households as a separate file</span> </span><span class="r"></span></p>
<p id="t509" class="run"><span class="n"><a href="#t509">509</a></span><span class="t"> <span class="nam">homes_by_uids</span> <span class="op">=</span> <span class="nam">homes_by_uids</span><span class="op">[</span><span class="nam">len</span><span class="op">(</span><span class="nam">facilities_by_uids</span><span class="op">)</span><span class="op">:</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t510" class="pln"><span class="n"><a href="#t510">510</a></span><span class="t"> <span class="com"># group uids to file</span> </span><span class="r"></span></p>
<p id="t511" class="run"><span class="n"><a href="#t511">511</a></span><span class="t"> <span class="key">if</span> <span class="nam">write</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t512" class="run"><span class="n"><a href="#t512">512</a></span><span class="t"> <span class="nam">sprw</span><span class="op">.</span><span class="nam">write_age_by_uid_dic</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">folder_name</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t513" class="pln"><span class="n"><a href="#t513">513</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t514" class="run"><span class="n"><a href="#t514">514</a></span><span class="t"> <span class="nam">sprw</span><span class="op">.</span><span class="nam">write_groups_by_age_and_uid</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">folder_name</span><span class="op">,</span> <span class="str">'households'</span><span class="op">,</span> <span class="nam">homes_by_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t515" class="run"><span class="n"><a href="#t515">515</a></span><span class="t"> <span class="nam">sprw</span><span class="op">.</span><span class="nam">write_groups_by_age_and_uid</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">folder_name</span><span class="op">,</span> <span class="str">'schools'</span><span class="op">,</span> <span class="nam">gen_school_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t516" class="run"><span class="n"><a href="#t516">516</a></span><span class="t"> <span class="nam">sprw</span><span class="op">.</span><span class="nam">write_groups_by_age_and_uid</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">folder_name</span><span class="op">,</span> <span class="str">'teachers'</span><span class="op">,</span> <span class="nam">gen_teacher_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t517" class="run"><span class="n"><a href="#t517">517</a></span><span class="t"> <span class="nam">sprw</span><span class="op">.</span><span class="nam">write_groups_by_age_and_uid</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">folder_name</span><span class="op">,</span> <span class="str">'workplaces'</span><span class="op">,</span> <span class="nam">gen_workplace_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t518" class="run"><span class="n"><a href="#t518">518</a></span><span class="t"> <span class="nam">sprw</span><span class="op">.</span><span class="nam">write_groups_by_age_and_uid</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">folder_name</span><span class="op">,</span> <span class="str">'facilities'</span><span class="op">,</span> <span class="nam">facilities_by_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t519" class="run"><span class="n"><a href="#t519">519</a></span><span class="t"> <span class="nam">sprw</span><span class="op">.</span><span class="nam">write_groups_by_age_and_uid</span><span class="op">(</span><span class="nam">datadir</span><span class="op">,</span> <span class="nam">location</span><span class="op">,</span> <span class="nam">state_location</span><span class="op">,</span> <span class="nam">country_location</span><span class="op">,</span> <span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">folder_name</span><span class="op">,</span> <span class="str">'facilities_staff'</span><span class="op">,</span> <span class="nam">facilities_staff_uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t520" class="pln"><span class="n"><a href="#t520">520</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t521" class="run"><span class="n"><a href="#t521">521</a></span><span class="t"> <span class="nam">popdict</span> <span class="op">=</span> <span class="nam">spct</span><span class="op">.</span><span class="nam">make_contacts_with_facilities_from_microstructure_objects</span><span class="op">(</span><span class="nam">age_by_uid_dic</span><span class="op">,</span> <span class="nam">homes_by_uids</span><span class="op">,</span> <span class="nam">gen_school_uids</span><span class="op">,</span> <span class="nam">gen_teacher_uids</span><span class="op">,</span> <span class="nam">gen_workplace_uids</span><span class="op">,</span> <span class="nam">facilities_by_uids</span><span class="op">,</span> <span class="nam">facilities_staff_uids</span><span class="op">,</span> <span class="nam">use_two_group_reduction</span><span class="op">=</span><span class="nam">use_two_group_reduction</span><span class="op">,</span> <span class="nam">average_LTCF_degree</span><span class="op">=</span><span class="nam">average_LTCF_degree</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t522" class="pln"><span class="n"><a href="#t522">522</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t523" class="run"><span class="n"><a href="#t523">523</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t524" class="mis show_mis"><span class="n"><a href="#t524">524</a></span><span class="t"> <span class="nam">uids</span> <span class="op">=</span> <span class="nam">popdict</span><span class="op">.</span><span class="nam">keys</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t525" class="mis show_mis"><span class="n"><a href="#t525">525</a></span><span class="t"> <span class="nam">uids</span> <span class="op">=</span> <span class="op">[</span><span class="nam">uid</span> <span class="key">for</span> <span class="nam">uid</span> <span class="key">in</span> <span class="nam">uids</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t526" class="mis show_mis"><span class="n"><a href="#t526">526</a></span><span class="t"> <span class="nam">np</span><span class="op">.</span><span class="nam">random</span><span class="op">.</span><span class="nam">shuffle</span><span class="op">(</span><span class="nam">uids</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t527" class="pln"><span class="n"><a href="#t527">527</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t528" class="mis show_mis"><span class="n"><a href="#t528">528</a></span><span class="t"> <span class="key">for</span> <span class="nam">i</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="num">20</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t529" class="mis show_mis"><span class="n"><a href="#t529">529</a></span><span class="t"> <span class="nam">uid</span> <span class="op">=</span> <span class="nam">uids</span><span class="op">[</span><span class="nam">i</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t530" class="mis show_mis"><span class="n"><a href="#t530">530</a></span><span class="t"> <span class="nam">person</span> <span class="op">=</span> <span class="nam">popdict</span><span class="op">[</span><span class="nam">uid</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t531" class="mis show_mis"><span class="n"><a href="#t531">531</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">uid</span><span class="op">,</span> <span class="nam">person</span><span class="op">[</span><span class="str">'age'</span><span class="op">]</span><span class="op">,</span> <span class="nam">person</span><span class="op">[</span><span class="str">'contacts'</span><span class="op">]</span><span class="op">[</span><span class="str">'H'</span><span class="op">]</span><span class="op">,</span> <span class="nam">person</span><span class="op">[</span><span class="str">'contacts'</span><span class="op">]</span><span class="op">[</span><span class="str">'S'</span><span class="op">]</span><span class="op">,</span> <span class="nam">person</span><span class="op">[</span><span class="str">'contacts'</span><span class="op">]</span><span class="op">[</span><span class="str">'W'</span><span class="op">]</span><span class="op">,</span> <span class="nam">person</span><span class="op">[</span><span class="str">'contacts'</span><span class="op">]</span><span class="op">[</span><span class="str">'LTCF'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t532" class="mis show_mis"><span class="n"><a href="#t532">532</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">person</span><span class="op">[</span><span class="str">'snf_res'</span><span class="op">]</span><span class="op">,</span> <span class="nam">person</span><span class="op">[</span><span class="str">'snf_staff'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t533" class="pln"><span class="n"><a href="#t533">533</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t534" class="run"><span class="n"><a href="#t534">534</a></span><span class="t"> <span class="key">if</span> <span class="nam">return_popdict</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t535" class="run"><span class="n"><a href="#t535">535</a></span><span class="t"> <span class="key">return</span> <span class="nam">popdict</span> </span><span class="r"></span></p>
<p id="t536" class="pln"><span class="n"><a href="#t536">536</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t537" class="pln"><span class="n"><a href="#t537">537</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t538" class="run"><span class="n"><a href="#t538">538</a></span><span class="t"><span class="key">def</span> <span class="nam">check_all_residents_are_connected_to_staff</span><span class="op">(</span><span class="nam">popdict</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t539" class="mis show_mis"><span class="n"><a href="#t539">539</a></span><span class="t"> <span class="nam">flag</span> <span class="op">=</span> <span class="key">True</span> </span><span class="r"></span></p>
<p id="t540" class="mis show_mis"><span class="n"><a href="#t540">540</a></span><span class="t"> <span class="key">for</span> <span class="nam">i</span> <span class="key">in</span> <span class="nam">popdict</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t541" class="mis show_mis"><span class="n"><a href="#t541">541</a></span><span class="t"> <span class="nam">person</span> <span class="op">=</span> <span class="nam">popdict</span><span class="op">[</span><span class="nam">i</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t542" class="mis show_mis"><span class="n"><a href="#t542">542</a></span><span class="t"> <span class="key">if</span> <span class="nam">person</span><span class="op">[</span><span class="str">'snf_res'</span><span class="op">]</span> <span class="op">==</span> <span class="num">1</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t543" class="pln"><span class="n"><a href="#t543">543</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t544" class="mis show_mis"><span class="n"><a href="#t544">544</a></span><span class="t"> <span class="nam">contacts</span> <span class="op">=</span> <span class="nam">person</span><span class="op">[</span><span class="str">'contacts'</span><span class="op">]</span><span class="op">[</span><span class="str">'LTCF'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t545" class="mis show_mis"><span class="n"><a href="#t545">545</a></span><span class="t"> <span class="nam">staff_contacts</span> <span class="op">=</span> <span class="op">[</span><span class="nam">j</span> <span class="key">for</span> <span class="nam">j</span> <span class="key">in</span> <span class="nam">contacts</span> <span class="key">if</span> <span class="nam">popdict</span><span class="op">[</span><span class="nam">j</span><span class="op">]</span><span class="op">[</span><span class="str">'snf_staff'</span><span class="op">]</span> <span class="op">==</span> <span class="num">1</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t546" class="pln"><span class="n"><a href="#t546">546</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t547" class="mis show_mis"><span class="n"><a href="#t547">547</a></span><span class="t"> <span class="key">if</span> <span class="nam">len</span><span class="op">(</span><span class="nam">staff_contacts</span><span class="op">)</span> <span class="op">==</span> <span class="num">0</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t548" class="mis show_mis"><span class="n"><a href="#t548">548</a></span><span class="t"> <span class="nam">flag</span> <span class="op">=</span> <span class="key">False</span> </span><span class="r"></span></p>
<p id="t549" class="mis show_mis"><span class="n"><a href="#t549">549</a></span><span class="t"> <span class="nam">errormsg</span> <span class="op">=</span> <span class="str">f'At least one LTCF or Skilled Nursing Facility resident has no contacts with staff members.'</span> </span><span class="r"></span></p>
<p id="t550" class="mis show_mis"><span class="n"><a href="#t550">550</a></span><span class="t"> <span class="key">raise</span> <span class="nam">ValueError</span><span class="op">(</span><span class="nam">errormsg</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t551" class="pln"><span class="n"><a href="#t551">551</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t552" class="mis show_mis"><span class="n"><a href="#t552">552</a></span><span class="t"> <span class="key">if</span> <span class="nam">flag</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t553" class="mis show_mis"><span class="n"><a href="#t553">553</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">'All LTCF residents have at least one contact with a staff member.'</span><span class="op">)</span> </span><span class="r"></span></p>
</div>
<div id="footer">
<div class="content">
<p>
<a class="nav" href="index.html">« index</a> <a class="nav" href="https://coverage.readthedocs.io">coverage.py v5.2</a>,
created at 2020-07-15 10:13
</p>
</div>
</div>
</body>
</html>