-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path_mnt_d_Github_synthpops_synthpops_config_py.html
183 lines (183 loc) · 34.7 KB
/
_mnt_d_Github_synthpops_synthpops_config_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
<!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 /mnt/d/Github/synthpops/synthpops/config.py: 63%</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>/mnt/d/Github/synthpops/synthpops/config.py</b> :
<span class="pc_cov">63%</span>
</h1>
<img id="keyboard_icon" src="keybd_closed.png" alt="Show keyboard shortcuts" />
<h2 class="stats">
63 statements
<button type="button" class="run shortkey_r button_toggle_run" title="Toggle lines run">40 run</button>
<button type="button" class="mis show_mis shortkey_m button_toggle_mis" title="Toggle lines missing">23 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">This module sets the location of the data folder and other global settings.</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">To change the level of log messages displayed, use e.g.</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="pln"><span class="n"><a href="#t6">6</a></span><span class="t"><span class="str"> sp.logger.setLevel('CRITICAL')</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 class="str">'''</span> </span><span class="r"></span></p>
<p id="t8" class="pln"><span class="n"><a href="#t8">8</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t9" class="pln"><span class="n"><a href="#t9">9</a></span><span class="t"><span class="com">#%% Housekeeping</span> </span><span class="r"></span></p>
<p id="t10" class="pln"><span class="n"><a href="#t10">10</a></span><span class="t"> </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">os</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">import</span> <span class="nam">sys</span> </span><span class="r"></span></p>
<p id="t13" class="run"><span class="n"><a href="#t13">13</a></span><span class="t"><span class="key">import</span> <span class="nam">psutil</span> </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">import</span> <span class="nam">sciris</span> <span class="key">as</span> <span class="nam">sc</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">import</span> <span class="nam">logging</span> </span><span class="r"></span></p>
<p id="t16" class="pln"><span class="n"><a href="#t16">16</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t17" class="pln"><span class="n"><a href="#t17">17</a></span><span class="t"> </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="nam">__all__</span> <span class="op">=</span> <span class="op">[</span><span class="str">'logger'</span><span class="op">,</span> <span class="str">'checkmem'</span><span class="op">,</span> <span class="str">'datadir'</span><span class="op">,</span> <span class="str">'localdatadir'</span><span class="op">,</span> <span class="str">'set_datadir'</span><span class="op">,</span> <span class="str">'set_nbrackets'</span><span class="op">,</span> <span class="str">'validate'</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t19" class="pln"><span class="n"><a href="#t19">19</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t20" class="pln"><span class="n"><a href="#t20">20</a></span><span class="t"><span class="com"># Declaring this here makes it globally available as synthpops.datadir</span> </span><span class="r"></span></p>
<p id="t21" class="run"><span class="n"><a href="#t21">21</a></span><span class="t"><span class="nam">datadir</span> <span class="op">=</span> <span class="key">None</span> </span><span class="r"></span></p>
<p id="t22" class="run"><span class="n"><a href="#t22">22</a></span><span class="t"><span class="nam">localdatadir</span> <span class="op">=</span> <span class="key">None</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="nam">full_data_available</span> <span class="op">=</span> <span class="key">False</span> <span class="com"># this is likely not necesary anymore</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><span class="r"></span></p>
<p id="t25" class="pln"><span class="n"><a href="#t25">25</a></span><span class="t"><span class="com"># Set the local data folder</span> </span><span class="r"></span></p>
<p id="t26" class="run"><span class="n"><a href="#t26">26</a></span><span class="t"><span class="nam">thisdir</span> <span class="op">=</span> <span class="nam">sc</span><span class="op">.</span><span class="nam">thisdir</span><span class="op">(</span><span class="nam">__file__</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t27" class="run"><span class="n"><a href="#t27">27</a></span><span class="t"><span class="nam">localdatadir</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">thisdir</span><span class="op">,</span> <span class="nam">os</span><span class="op">.</span><span class="nam">pardir</span><span class="op">,</span> <span class="str">'data'</span><span class="op">)</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><span class="r"></span></p>
<p id="t29" class="pln"><span class="n"><a href="#t29">29</a></span><span class="t"><span class="com"># Replace with local data dir if Dropbox folder is not found</span> </span><span class="r"></span></p>
<p id="t30" class="run"><span class="n"><a href="#t30">30</a></span><span class="t"><span class="key">if</span> <span class="nam">datadir</span> <span class="key">is</span> <span class="key">None</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t31" class="run"><span class="n"><a href="#t31">31</a></span><span class="t"> <span class="nam">full_data_available</span> <span class="op">=</span> <span class="key">True</span> </span><span class="r"></span></p>
<p id="t32" class="run"><span class="n"><a href="#t32">32</a></span><span class="t"> <span class="nam">datadir</span> <span class="op">=</span> <span class="nam">localdatadir</span> </span><span class="r"></span></p>
<p id="t33" class="pln"><span class="n"><a href="#t33">33</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t34" class="pln"><span class="n"><a href="#t34">34</a></span><span class="t"> </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="com"># Number of census age brackets to use</span> </span><span class="r"></span></p>
<p id="t36" class="run"><span class="n"><a href="#t36">36</a></span><span class="t"><span class="nam">nbrackets</span> <span class="op">=</span> <span class="op">[</span><span class="num">16</span><span class="op">,</span> <span class="num">20</span><span class="op">]</span><span class="op">[</span><span class="num">1</span><span class="op">]</span> <span class="com"># Choose how many age bins to use -- 20 is only partially supported</span> </span><span class="r"></span></p>
<p id="t37" class="run"><span class="n"><a href="#t37">37</a></span><span class="t"><span class="nam">matrix_size</span> <span class="op">=</span> <span class="num">16</span> <span class="com"># The dimensions of the mixing matrices -- currently only 16 is available</span> </span><span class="r"></span></p>
<p id="t38" class="pln"><span class="n"><a href="#t38">38</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t39" class="pln"><span class="n"><a href="#t39">39</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t40" class="pln"><span class="n"><a href="#t40">40</a></span><span class="t"><span class="com">#%% Logger -- adapted from Atomica</span> </span><span class="r"></span></p>
<p id="t41" class="pln"><span class="n"><a href="#t41">41</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t42" class="pln"><span class="n"><a href="#t42">42</a></span><span class="t"><span class="com"># Set the default logging level</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="nam">default_log_level</span> <span class="op">=</span> <span class="op">[</span><span class="str">'DEBUG'</span><span class="op">,</span> <span class="str">'INFO'</span><span class="op">,</span> <span class="str">'WARNING'</span><span class="op">,</span> <span class="str">'CRITICAL'</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="t44" class="pln"><span class="n"><a href="#t44">44</a></span><span class="t"> </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="nam">logger</span> <span class="op">=</span> <span class="nam">logging</span><span class="op">.</span><span class="nam">getLogger</span><span class="op">(</span><span class="str">'synthpops'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t46" class="pln"><span class="n"><a href="#t46">46</a></span><span class="t"> </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="key">if</span> <span class="key">not</span> <span class="nam">logger</span><span class="op">.</span><span class="nam">hasHandlers</span><span class="op">(</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t48" class="pln"><span class="n"><a href="#t48">48</a></span><span class="t"> <span class="com"># Only add handlers if they don't already exist in the module-level logger</span> </span><span class="r"></span></p>
<p id="t49" class="pln"><span class="n"><a href="#t49">49</a></span><span class="t"> <span class="com"># This means that it's possible for the user to completely customize *a* logger called 'atomica'</span> </span><span class="r"></span></p>
<p id="t50" class="pln"><span class="n"><a href="#t50">50</a></span><span class="t"> <span class="com"># prior to importing Atomica, and the user's custom logger won't be overwritten as long as it has</span> </span><span class="r"></span></p>
<p id="t51" class="pln"><span class="n"><a href="#t51">51</a></span><span class="t"> <span class="com"># at least one handler already added. The use case was originally to suppress messages on import, but since</span> </span><span class="r"></span></p>
<p id="t52" class="pln"><span class="n"><a href="#t52">52</a></span><span class="t"> <span class="com"># importing is silent now, it doesn't matter so much.</span> </span><span class="r"></span></p>
<p id="t53" class="mis show_mis"><span class="n"><a href="#t53">53</a></span><span class="t"> <span class="nam">debug_handler</span> <span class="op">=</span> <span class="nam">logging</span><span class="op">.</span><span class="nam">StreamHandler</span><span class="op">(</span><span class="nam">sys</span><span class="op">.</span><span class="nam">stdout</span><span class="op">)</span> <span class="com"># info_handler will handle all messages below WARNING sending them to STDOUT</span> </span><span class="r"></span></p>
<p id="t54" class="mis show_mis"><span class="n"><a href="#t54">54</a></span><span class="t"> <span class="nam">info_handler</span> <span class="op">=</span> <span class="nam">logging</span><span class="op">.</span><span class="nam">StreamHandler</span><span class="op">(</span><span class="nam">sys</span><span class="op">.</span><span class="nam">stdout</span><span class="op">)</span> <span class="com"># info_handler will handle all messages below WARNING sending them to STDOUT</span> </span><span class="r"></span></p>
<p id="t55" class="mis show_mis"><span class="n"><a href="#t55">55</a></span><span class="t"> <span class="nam">warning_handler</span> <span class="op">=</span> <span class="nam">logging</span><span class="op">.</span><span class="nam">StreamHandler</span><span class="op">(</span><span class="nam">sys</span><span class="op">.</span><span class="nam">stderr</span><span class="op">)</span> <span class="com"># warning_handler will send all messages at or above WARNING to STDERR</span> </span><span class="r"></span></p>
<p id="t56" class="pln"><span class="n"><a href="#t56">56</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t57" class="pln"><span class="n"><a href="#t57">57</a></span><span class="t"> <span class="com"># Handle levels</span> </span><span class="r"></span></p>
<p id="t58" class="mis show_mis"><span class="n"><a href="#t58">58</a></span><span class="t"> <span class="nam">debug_handler</span><span class="op">.</span><span class="nam">setLevel</span><span class="op">(</span><span class="num">0</span><span class="op">)</span> <span class="com"># Handle all lower levels - the output should be filtered further by setting the logger level, not the handler level</span> </span><span class="r"></span></p>
<p id="t59" class="mis show_mis"><span class="n"><a href="#t59">59</a></span><span class="t"> <span class="nam">info_handler</span><span class="op">.</span><span class="nam">setLevel</span><span class="op">(</span><span class="nam">logging</span><span class="op">.</span><span class="nam">INFO</span><span class="op">)</span> <span class="com"># Handle all lower levels - the output should be filtered further by setting the logger level, not the handler level</span> </span><span class="r"></span></p>
<p id="t60" class="mis show_mis"><span class="n"><a href="#t60">60</a></span><span class="t"> <span class="nam">warning_handler</span><span class="op">.</span><span class="nam">setLevel</span><span class="op">(</span><span class="nam">logging</span><span class="op">.</span><span class="nam">WARNING</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t61" class="mis show_mis"><span class="n"><a href="#t61">61</a></span><span class="t"> <span class="nam">debug_handler</span><span class="op">.</span><span class="nam">addFilter</span><span class="op">(</span><span class="nam">type</span><span class="op">(</span><span class="str">"ThresholdFilter"</span><span class="op">,</span> <span class="op">(</span><span class="nam">object</span><span class="op">,</span><span class="op">)</span><span class="op">,</span> <span class="op">{</span><span class="str">"filter"</span><span class="op">:</span> <span class="key">lambda</span> <span class="nam">x</span><span class="op">,</span> <span class="nam">logRecord</span><span class="op">:</span> <span class="nam">logRecord</span><span class="op">.</span><span class="nam">levelno</span> <span class="op"><</span> <span class="nam">logging</span><span class="op">.</span><span class="nam">INFO</span><span class="op">}</span><span class="op">)</span><span class="op">(</span><span class="op">)</span><span class="op">)</span> <span class="com"># Display anything INFO or higher</span> </span><span class="r"></span></p>
<p id="t62" class="mis show_mis"><span class="n"><a href="#t62">62</a></span><span class="t"> <span class="nam">info_handler</span><span class="op">.</span><span class="nam">addFilter</span><span class="op">(</span><span class="nam">type</span><span class="op">(</span><span class="str">"ThresholdFilter"</span><span class="op">,</span> <span class="op">(</span><span class="nam">object</span><span class="op">,</span><span class="op">)</span><span class="op">,</span> <span class="op">{</span><span class="str">"filter"</span><span class="op">:</span> <span class="key">lambda</span> <span class="nam">x</span><span class="op">,</span> <span class="nam">logRecord</span><span class="op">:</span> <span class="nam">logRecord</span><span class="op">.</span><span class="nam">levelno</span> <span class="op"><</span> <span class="nam">logging</span><span class="op">.</span><span class="nam">WARNING</span><span class="op">}</span><span class="op">)</span><span class="op">(</span><span class="op">)</span><span class="op">)</span> <span class="com"># Don't display WARNING or higher</span> </span><span class="r"></span></p>
<p id="t63" class="pln"><span class="n"><a href="#t63">63</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t64" class="pln"><span class="n"><a href="#t64">64</a></span><span class="t"> <span class="com"># Set formatting and log level</span> </span><span class="r"></span></p>
<p id="t65" class="mis show_mis"><span class="n"><a href="#t65">65</a></span><span class="t"> <span class="nam">formatter</span> <span class="op">=</span> <span class="nam">logging</span><span class="op">.</span><span class="nam">Formatter</span><span class="op">(</span><span class="str">'%(levelname)s %(asctime)s.%(msecs)d {%(filename)s:%(lineno)d} - %(message)s'</span><span class="op">,</span> <span class="nam">datefmt</span><span class="op">=</span><span class="str">'%H:%M:%S'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t66" class="mis show_mis"><span class="n"><a href="#t66">66</a></span><span class="t"> <span class="nam">debug_handler</span><span class="op">.</span><span class="nam">setFormatter</span><span class="op">(</span><span class="nam">formatter</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t67" class="mis show_mis"><span class="n"><a href="#t67">67</a></span><span class="t"> <span class="key">for</span> <span class="nam">handler</span> <span class="key">in</span> <span class="op">[</span><span class="nam">debug_handler</span><span class="op">,</span> <span class="nam">info_handler</span><span class="op">,</span> <span class="nam">warning_handler</span><span class="op">]</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t68" class="mis show_mis"><span class="n"><a href="#t68">68</a></span><span class="t"> <span class="nam">logger</span><span class="op">.</span><span class="nam">addHandler</span><span class="op">(</span><span class="nam">handler</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t69" class="mis show_mis"><span class="n"><a href="#t69">69</a></span><span class="t"> <span class="nam">logger</span><span class="op">.</span><span class="nam">setLevel</span><span class="op">(</span><span class="nam">default_log_level</span><span class="op">)</span> <span class="com"># Set the overall log level</span> </span><span class="r"></span></p>
<p id="t70" class="pln"><span class="n"><a href="#t70">70</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t71" class="pln"><span class="n"><a href="#t71">71</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t72" class="run"><span class="n"><a href="#t72">72</a></span><span class="t"><span class="key">def</span> <span class="nam">checkmem</span><span class="op">(</span><span class="nam">unit</span><span class="op">=</span><span class="str">'mb'</span><span class="op">,</span> <span class="nam">fmt</span><span class="op">=</span><span class="str">'0.2f'</span><span class="op">,</span> <span class="nam">start</span><span class="op">=</span><span class="num">0</span><span class="op">,</span> <span class="nam">to_string</span><span class="op">=</span><span class="key">True</span><span class="op">)</span><span class="op">:</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">''' For use with logger, check current memory usage '''</span> </span><span class="r"></span></p>
<p id="t74" class="run"><span class="n"><a href="#t74">74</a></span><span class="t"> <span class="nam">process</span> <span class="op">=</span> <span class="nam">psutil</span><span class="op">.</span><span class="nam">Process</span><span class="op">(</span><span class="nam">os</span><span class="op">.</span><span class="nam">getpid</span><span class="op">(</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t75" class="run"><span class="n"><a href="#t75">75</a></span><span class="t"> <span class="nam">mapping</span> <span class="op">=</span> <span class="op">{</span><span class="str">'b'</span><span class="op">:</span><span class="num">1</span><span class="op">,</span> <span class="str">'kb'</span><span class="op">:</span><span class="num">1e3</span><span class="op">,</span> <span class="str">'mb'</span><span class="op">:</span><span class="num">1e6</span><span class="op">,</span> <span class="str">'gb'</span><span class="op">:</span><span class="num">1e9</span><span class="op">}</span> </span><span class="r"></span></p>
<p id="t76" class="run"><span class="n"><a href="#t76">76</a></span><span class="t"> <span class="key">try</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t77" class="run"><span class="n"><a href="#t77">77</a></span><span class="t"> <span class="nam">factor</span> <span class="op">=</span> <span class="nam">mapping</span><span class="op">[</span><span class="nam">unit</span><span class="op">.</span><span class="nam">lower</span><span class="op">(</span><span class="op">)</span><span class="op">]</span> </span><span class="r"></span></p>
<p id="t78" class="mis show_mis"><span class="n"><a href="#t78">78</a></span><span class="t"> <span class="key">except</span> <span class="nam">KeyError</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t79" class="mis show_mis"><span class="n"><a href="#t79">79</a></span><span class="t"> <span class="key">raise</span> <span class="nam">sc</span><span class="op">.</span><span class="nam">KeyNotFoundError</span><span class="op">(</span><span class="str">f'Unit {unit} not found'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t80" class="run"><span class="n"><a href="#t80">80</a></span><span class="t"> <span class="nam">mem_use</span> <span class="op">=</span> <span class="nam">process</span><span class="op">.</span><span class="nam">memory_info</span><span class="op">(</span><span class="op">)</span><span class="op">.</span><span class="nam">rss</span><span class="op">/</span><span class="nam">factor</span> <span class="op">-</span> <span class="nam">start</span> </span><span class="r"></span></p>
<p id="t81" class="run"><span class="n"><a href="#t81">81</a></span><span class="t"> <span class="key">if</span> <span class="nam">to_string</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t82" class="run"><span class="n"><a href="#t82">82</a></span><span class="t"> <span class="nam">output</span> <span class="op">=</span> <span class="str">f'{mem_use:{fmt}} {unit.upper()}'</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="key">else</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t84" class="mis show_mis"><span class="n"><a href="#t84">84</a></span><span class="t"> <span class="nam">output</span> <span class="op">=</span> <span class="nam">mem_use</span> </span><span class="r"></span></p>
<p id="t85" class="run"><span class="n"><a href="#t85">85</a></span><span class="t"> <span class="key">return</span> <span class="nam">output</span> </span><span class="r"></span></p>
<p id="t86" class="pln"><span class="n"><a href="#t86">86</a></span><span class="t"> </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="pln"><span class="n"><a href="#t88">88</a></span><span class="t"><span class="com">#%% Functions</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">logger</span><span class="op">.</span><span class="nam">info</span><span class="op">(</span><span class="str">f'Loading SynthPops: {thisdir}'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t91" class="run"><span class="n"><a href="#t91">91</a></span><span class="t"><span class="nam">logger</span><span class="op">.</span><span class="nam">debug</span><span class="op">(</span><span class="str">f'Data folder: {datadir}'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t92" class="pln"><span class="n"><a href="#t92">92</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t93" class="run"><span class="n"><a href="#t93">93</a></span><span class="t"><span class="key">def</span> <span class="nam">set_datadir</span><span class="op">(</span><span class="nam">folder</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t94" class="pln"><span class="n"><a href="#t94">94</a></span><span class="t"> <span class="str">'''Set the data folder to the user-specified location -- note, mostly deprecated.'''</span> </span><span class="r"></span></p>
<p id="t95" class="pln"><span class="n"><a href="#t95">95</a></span><span class="t"> <span class="key">global</span> <span class="nam">datadir</span> </span><span class="r"></span></p>
<p id="t96" class="mis show_mis"><span class="n"><a href="#t96">96</a></span><span class="t"> <span class="nam">datadir</span> <span class="op">=</span> <span class="nam">folder</span> </span><span class="r"></span></p>
<p id="t97" class="mis show_mis"><span class="n"><a href="#t97">97</a></span><span class="t"> <span class="nam">logger</span><span class="op">.</span><span class="nam">info</span><span class="op">(</span><span class="str">f'Done: data directory set to {folder}.'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t98" class="mis show_mis"><span class="n"><a href="#t98">98</a></span><span class="t"> <span class="key">return</span> <span class="nam">datadir</span> </span><span class="r"></span></p>
<p id="t99" class="pln"><span class="n"><a href="#t99">99</a></span><span class="t"> </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="key">def</span> <span class="nam">set_nbrackets</span><span class="op">(</span><span class="nam">n</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 class="str">'''Set the number of census brackets -- usually 16 or 20.'''</span> </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="key">global</span> <span class="nam">nbrackets</span> </span><span class="r"></span></p>
<p id="t103" class="run"><span class="n"><a href="#t103">103</a></span><span class="t"> <span class="nam">nbrackets</span> <span class="op">=</span> <span class="nam">n</span> </span><span class="r"></span></p>
<p id="t104" class="run"><span class="n"><a href="#t104">104</a></span><span class="t"> <span class="key">if</span> <span class="nam">nbrackets</span> <span class="key">not</span> <span class="key">in</span> <span class="op">[</span><span class="num">16</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="t105" class="mis show_mis"><span class="n"><a href="#t105">105</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="str">f'Note: current supported bracket choices are 16 or 20, use {nbrackets} at your own risk.'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t106" class="run"><span class="n"><a href="#t106">106</a></span><span class="t"> <span class="nam">logger</span><span class="op">.</span><span class="nam">info</span><span class="op">(</span><span class="str">f'Done: number of brackets is set to {n}.'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t107" class="run"><span class="n"><a href="#t107">107</a></span><span class="t"> <span class="key">return</span> <span class="nam">nbrackets</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><span class="r"></span></p>
<p id="t109" class="pln"><span class="n"><a href="#t109">109</a></span><span class="t"> </span><span class="r"></span></p>
<p id="t110" class="run"><span class="n"><a href="#t110">110</a></span><span class="t"><span class="key">def</span> <span class="nam">validate</span><span class="op">(</span><span class="nam">verbose</span><span class="op">=</span><span class="key">True</span><span class="op">)</span><span class="op">:</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">''' Check that the data folder can be found. '''</span> </span><span class="r"></span></p>
<p id="t112" class="run"><span class="n"><a href="#t112">112</a></span><span class="t"> <span class="key">if</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">isdir</span><span class="op">(</span><span class="nam">datadir</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t113" class="run"><span class="n"><a href="#t113">113</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="t114" class="run"><span class="n"><a href="#t114">114</a></span><span class="t"> <span class="nam">logger</span><span class="op">.</span><span class="nam">debug</span><span class="op">(</span><span class="str">f'The data folder {datadir} was found.'</span><span class="op">)</span> </span><span class="r"></span></p>
<p id="t115" class="pln"><span class="n"><a href="#t115">115</a></span><span class="t"> <span class="key">else</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t116" class="mis show_mis"><span class="n"><a href="#t116">116</a></span><span class="t"> <span class="key">if</span> <span class="nam">datadir</span> <span class="key">is</span> <span class="key">None</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t117" class="mis show_mis"><span class="n"><a href="#t117">117</a></span><span class="t"> <span class="key">raise</span> <span class="nam">FileNotFoundError</span><span class="op">(</span><span class="str">f'The datadir has not been set; use synthpops.set_datadir() and try again.'</span><span class="op">)</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 class="key">else</span><span class="op">:</span> </span><span class="r"></span></p>
<p id="t119" class="mis show_mis"><span class="n"><a href="#t119">119</a></span><span class="t"> <span class="key">raise</span> <span class="nam">FileNotFoundError</span><span class="op">(</span><span class="str">f'The folder "{datadir}" does not exist, as far as I can tell.'</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.3</a>,
created at 2020-10-05 17:01 -0700
</p>
</div>
</div>
</body>
</html>