Spaces:
Running
Running
Commit
Β·
52580dc
1
Parent(s):
7e51632
edit the docs
Browse files- README.md +202 -20
- pyproject.toml +2 -0
- tests/test_dataloading/test_dataingestion.py +189 -0
README.md
CHANGED
@@ -18,34 +18,216 @@ StingrayExplorer is a data analysis tool designed for quick visualization and ex
|
|
18 |
|
19 |
```
|
20 |
.
|
|
|
|
|
|
|
|
|
21 |
βββ assets
|
22 |
-
β
|
23 |
-
β
|
24 |
-
β
|
25 |
-
β
|
26 |
-
β
|
27 |
-
β
|
28 |
-
β
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
βββ explorer.py
|
30 |
βββ files
|
31 |
-
β
|
32 |
-
β
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
βββ modules
|
34 |
-
β
|
35 |
-
β
|
36 |
-
β
|
37 |
-
β
|
38 |
-
β
|
39 |
-
β
|
40 |
-
β
|
41 |
-
β
|
42 |
-
β
|
43 |
-
β
|
44 |
-
βββ
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
45 |
βββ DashboardClasses.py
|
|
|
|
|
|
|
|
|
|
|
46 |
βββ globals.py
|
47 |
βββ sidebar.py
|
48 |
βββ strings.py
|
|
|
|
|
49 |
```
|
50 |
|
51 |
## Installation
|
|
|
18 |
|
19 |
```
|
20 |
.
|
21 |
+
βββ CODE_OF_CONDUCT.md
|
22 |
+
βββ Dockerfile
|
23 |
+
βββ LICENSE
|
24 |
+
βββ README.md
|
25 |
βββ assets
|
26 |
+
βΒ Β βββ audio
|
27 |
+
βΒ Β βββ icons
|
28 |
+
βΒ Β βΒ Β βββ __pycache__
|
29 |
+
βΒ Β βΒ Β βΒ Β βββ svg.cpython-311.pyc
|
30 |
+
βΒ Β βΒ Β βββ svg.py
|
31 |
+
βΒ Β βββ images
|
32 |
+
βΒ Β βΒ Β βββ holoviz_logo.png
|
33 |
+
βΒ Β βΒ Β βββ holoviz_logo_minimised.png
|
34 |
+
βΒ Β βΒ Β βββ stingray_explorer.png
|
35 |
+
βΒ Β βΒ Β βββ stingray_explorer.webp
|
36 |
+
βΒ Β βΒ Β βββ stingray_explorer_minimised.png
|
37 |
+
βΒ Β βΒ Β βββ stingray_logo.png
|
38 |
+
βΒ Β βΒ Β βββ stingray_logo_minimised.png
|
39 |
+
βΒ Β βββ stylesheets
|
40 |
+
βΒ Β βΒ Β βββ explorer.css
|
41 |
+
βΒ Β βββ videos
|
42 |
+
βββ docs
|
43 |
+
βΒ Β βββ Makefile
|
44 |
+
βΒ Β βββ build
|
45 |
+
βΒ Β βΒ Β βββ doctrees
|
46 |
+
βΒ Β βΒ Β βΒ Β βββ DataLoading.doctree
|
47 |
+
βΒ Β βΒ Β βΒ Β βββ Home.doctree
|
48 |
+
βΒ Β βΒ Β βΒ Β βββ QuickLook.doctree
|
49 |
+
βΒ Β βΒ Β βΒ Β βββ environment.pickle
|
50 |
+
βΒ Β βΒ Β βΒ Β βββ index.doctree
|
51 |
+
βΒ Β βΒ Β βΒ Β βββ modules.doctree
|
52 |
+
βΒ Β βΒ Β βββ html
|
53 |
+
βΒ Β βΒ Β βββ DataLoading.html
|
54 |
+
βΒ Β βΒ Β βββ Home.html
|
55 |
+
βΒ Β βΒ Β βββ QuickLook.html
|
56 |
+
βΒ Β βΒ Β βββ _modules
|
57 |
+
βΒ Β βΒ Β βΒ Β βββ index.html
|
58 |
+
βΒ Β βΒ Β βΒ Β βββ modules
|
59 |
+
βΒ Β βΒ Β βΒ Β βββ DataLoading
|
60 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ DataIngestion.html
|
61 |
+
βΒ Β βΒ Β βΒ Β βββ Home
|
62 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ HomeContent.html
|
63 |
+
βΒ Β βΒ Β βΒ Β βββ QuickLook
|
64 |
+
βΒ Β βΒ Β βΒ Β βββ AverageCrossSpectrum.html
|
65 |
+
βΒ Β βΒ Β βΒ Β βββ AveragePowerSpectrum.html
|
66 |
+
βΒ Β βΒ Β βΒ Β βββ Bispectrum.html
|
67 |
+
βΒ Β βΒ Β βΒ Β βββ CrossSpectrum.html
|
68 |
+
βΒ Β βΒ Β βΒ Β βββ LightCurve.html
|
69 |
+
βΒ Β βΒ Β βΒ Β βββ PowerColors.html
|
70 |
+
βΒ Β βΒ Β βΒ Β βββ PowerSpectrum.html
|
71 |
+
βΒ Β βΒ Β βββ _sources
|
72 |
+
βΒ Β βΒ Β βΒ Β βββ DataLoading.rst.txt
|
73 |
+
βΒ Β βΒ Β βΒ Β βββ Home.rst.txt
|
74 |
+
βΒ Β βΒ Β βΒ Β βββ QuickLook.rst.txt
|
75 |
+
βΒ Β βΒ Β βΒ Β βββ index.rst.txt
|
76 |
+
βΒ Β βΒ Β βΒ Β βββ modules.rst.txt
|
77 |
+
βΒ Β βΒ Β βββ _static
|
78 |
+
βΒ Β βΒ Β βΒ Β βββ _sphinx_javascript_frameworks_compat.js
|
79 |
+
βΒ Β βΒ Β βΒ Β βββ basic.css
|
80 |
+
βΒ Β βΒ Β βΒ Β βββ css
|
81 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ badge_only.css
|
82 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ fonts
|
83 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ Roboto-Slab-Bold.woff
|
84 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ Roboto-Slab-Bold.woff2
|
85 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ Roboto-Slab-Regular.woff
|
86 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ Roboto-Slab-Regular.woff2
|
87 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ fontawesome-webfont.eot
|
88 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ fontawesome-webfont.svg
|
89 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ fontawesome-webfont.ttf
|
90 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ fontawesome-webfont.woff
|
91 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ fontawesome-webfont.woff2
|
92 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold-italic.woff
|
93 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold-italic.woff2
|
94 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold.woff
|
95 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold.woff2
|
96 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-normal-italic.woff
|
97 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-normal-italic.woff2
|
98 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-normal.woff
|
99 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-normal.woff2
|
100 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ theme.css
|
101 |
+
βΒ Β βΒ Β βΒ Β βββ doctools.js
|
102 |
+
βΒ Β βΒ Β βΒ Β βββ documentation_options.js
|
103 |
+
βΒ Β βΒ Β βΒ Β βββ file.png
|
104 |
+
βΒ Β βΒ Β βΒ Β βββ fonts
|
105 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ Lato
|
106 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold.eot
|
107 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold.ttf
|
108 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold.woff
|
109 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bold.woff2
|
110 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bolditalic.eot
|
111 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bolditalic.ttf
|
112 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bolditalic.woff
|
113 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-bolditalic.woff2
|
114 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-italic.eot
|
115 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-italic.ttf
|
116 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-italic.woff
|
117 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-italic.woff2
|
118 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-regular.eot
|
119 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-regular.ttf
|
120 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-regular.woff
|
121 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βΒ Β βββ lato-regular.woff2
|
122 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ RobotoSlab
|
123 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-bold.eot
|
124 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-bold.ttf
|
125 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-bold.woff
|
126 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-bold.woff2
|
127 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-regular.eot
|
128 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-regular.ttf
|
129 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-regular.woff
|
130 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ roboto-slab-v7-regular.woff2
|
131 |
+
βΒ Β βΒ Β βΒ Β βββ jquery.js
|
132 |
+
βΒ Β βΒ Β βΒ Β βββ js
|
133 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ badge_only.js
|
134 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ theme.js
|
135 |
+
βΒ Β βΒ Β βΒ Β βΒ Β βββ versions.js
|
136 |
+
βΒ Β βΒ Β βΒ Β βββ language_data.js
|
137 |
+
βΒ Β βΒ Β βΒ Β βββ minus.png
|
138 |
+
βΒ Β βΒ Β βΒ Β βββ plus.png
|
139 |
+
βΒ Β βΒ Β βΒ Β βββ pygments.css
|
140 |
+
βΒ Β βΒ Β βΒ Β βββ searchtools.js
|
141 |
+
βΒ Β βΒ Β βΒ Β βββ sphinx_highlight.js
|
142 |
+
βΒ Β βΒ Β βββ genindex.html
|
143 |
+
βΒ Β βΒ Β βββ index.html
|
144 |
+
βΒ Β βΒ Β βββ modules.html
|
145 |
+
βΒ Β βΒ Β βββ objects.inv
|
146 |
+
βΒ Β βΒ Β βββ py-modindex.html
|
147 |
+
βΒ Β βΒ Β βββ search.html
|
148 |
+
βΒ Β βΒ Β βββ searchindex.js
|
149 |
+
βΒ Β βββ files
|
150 |
+
βΒ Β βΒ Β βββ loaded-data
|
151 |
+
βΒ Β βββ make.bat
|
152 |
+
βΒ Β βββ requirements.txt
|
153 |
+
βΒ Β βββ source
|
154 |
+
βΒ Β βββ DataLoading.rst
|
155 |
+
βΒ Β βββ Home.rst
|
156 |
+
βΒ Β βββ QuickLook.rst
|
157 |
+
βΒ Β βββ _static
|
158 |
+
βΒ Β βββ _templates
|
159 |
+
βΒ Β βββ conf.py
|
160 |
+
βΒ Β βββ index.rst
|
161 |
+
βΒ Β βββ modules.rst
|
162 |
+
βββ environment.yml
|
163 |
βββ explorer.py
|
164 |
βββ files
|
165 |
+
βΒ Β βββ data
|
166 |
+
βΒ Β βΒ Β βββ LightCurve_bexvar.fits
|
167 |
+
βΒ Β βΒ Β βββ data_small.hdf5
|
168 |
+
βΒ Β βΒ Β βββ data_smaller.hdf5
|
169 |
+
βΒ Β βΒ Β βββ lcurveA.fits
|
170 |
+
βΒ Β βΒ Β βββ lcurve_new.fits
|
171 |
+
βΒ Β βΒ Β βββ monol_testA.evt
|
172 |
+
βΒ Β βΒ Β βββ monol_testA_calib.evt
|
173 |
+
βΒ Β βΒ Β βββ monol_testA_calib_unsrt.evt
|
174 |
+
βΒ Β βΒ Β βββ nomission.evt
|
175 |
+
βΒ Β βΒ Β βββ xte_gx_test.evt.gz
|
176 |
+
βΒ Β βΒ Β βββ xte_test.evt.gz
|
177 |
+
βΒ Β βββ loaded-data
|
178 |
βββ modules
|
179 |
+
βΒ Β βββ DataLoading
|
180 |
+
βΒ Β βΒ Β βββ DataIngestion.py
|
181 |
+
βΒ Β βΒ Β βββ __init__.py
|
182 |
+
βΒ Β βΒ Β βββ __pycache__
|
183 |
+
βΒ Β βΒ Β βββ DataIngestion.cpython-311.pyc
|
184 |
+
βΒ Β βΒ Β βββ __init__.cpython-311.pyc
|
185 |
+
βΒ Β βββ Home
|
186 |
+
βΒ Β βΒ Β βββ HomeContent.py
|
187 |
+
βΒ Β βΒ Β βββ __init__.py
|
188 |
+
βΒ Β βΒ Β βββ __pycache__
|
189 |
+
βΒ Β βΒ Β βββ Home.cpython-311.pyc
|
190 |
+
βΒ Β βΒ Β βββ HomeContent.cpython-311.pyc
|
191 |
+
βΒ Β βΒ Β βββ __init__.cpython-311.pyc
|
192 |
+
βΒ Β βββ QuickLook
|
193 |
+
βΒ Β βββ AverageCrossSpectrum.py
|
194 |
+
βΒ Β βββ AveragePowerSpectrum.py
|
195 |
+
βΒ Β βββ Bispectrum.py
|
196 |
+
βΒ Β βββ CrossSpectrum.py
|
197 |
+
βΒ Β βββ LightCurve.py
|
198 |
+
βΒ Β βββ PowerColors.py
|
199 |
+
βΒ Β βββ PowerSpectrum.py
|
200 |
+
βΒ Β βββ __init__.py
|
201 |
+
βΒ Β βββ __pycache__
|
202 |
+
βΒ Β βββ AverageCrossSpectrum.cpython-311.pyc
|
203 |
+
βΒ Β βββ AveragePowerSpectrum.cpython-311.pyc
|
204 |
+
βΒ Β βββ Bispectrum.cpython-311.pyc
|
205 |
+
βΒ Β βββ CrossSpectrum.cpython-311.pyc
|
206 |
+
βΒ Β βββ LightCurve.cpython-311.pyc
|
207 |
+
βΒ Β βββ PowerColors.cpython-311.pyc
|
208 |
+
βΒ Β βββ PowerSpectrum.cpython-311.pyc
|
209 |
+
βΒ Β βββ __init__.cpython-311.pyc
|
210 |
+
βββ pyproject.toml
|
211 |
+
βββ setup.py
|
212 |
+
βββ tests
|
213 |
+
βΒ Β βββ test_dataloading
|
214 |
+
βΒ Β βΒ Β βββ __pycache__
|
215 |
+
βΒ Β βΒ Β βΒ Β βββ test_dataingestion.cpython-311-pytest-8.2.1.pyc
|
216 |
+
βΒ Β βΒ Β βββ test_dataingestion.py
|
217 |
+
βΒ Β βββ test_quicklook
|
218 |
+
βΒ Β βββ test_utils
|
219 |
+
βββ utils
|
220 |
βββ DashboardClasses.py
|
221 |
+
βββ __pycache__
|
222 |
+
βΒ Β βββ dashboardClasses.cpython-311.pyc
|
223 |
+
βΒ Β βββ globals.cpython-311.pyc
|
224 |
+
βΒ Β βββ sidebar.cpython-311.pyc
|
225 |
+
βΒ Β βββ strings.cpython-311.pyc
|
226 |
βββ globals.py
|
227 |
βββ sidebar.py
|
228 |
βββ strings.py
|
229 |
+
|
230 |
+
47 directories, 162 files
|
231 |
```
|
232 |
|
233 |
## Installation
|
pyproject.toml
CHANGED
@@ -14,3 +14,5 @@ dependencies = [
|
|
14 |
"stingray",
|
15 |
# Add other dependencies here if needed
|
16 |
]
|
|
|
|
|
|
14 |
"stingray",
|
15 |
# Add other dependencies here if needed
|
16 |
]
|
17 |
+
[tool.pytest.ini_options]
|
18 |
+
pythonpath = ["."]
|
tests/test_dataloading/test_dataingestion.py
ADDED
@@ -0,0 +1,189 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import pytest
|
2 |
+
from unittest.mock import MagicMock, patch
|
3 |
+
from modules.DataLoading.DataIngestion import (
|
4 |
+
create_loadingdata_output_box,
|
5 |
+
load_event_data,
|
6 |
+
save_loaded_files,
|
7 |
+
delete_selected_files,
|
8 |
+
preview_loaded_files,
|
9 |
+
clear_loaded_files,
|
10 |
+
create_event_list,
|
11 |
+
simulate_event_list,
|
12 |
+
create_warning_handler,
|
13 |
+
)
|
14 |
+
|
15 |
+
|
16 |
+
def test_create_loadingdata_output_box():
|
17 |
+
content = "File loaded successfully."
|
18 |
+
output_box = create_loadingdata_output_box(content)
|
19 |
+
assert output_box.output_content == content
|
20 |
+
|
21 |
+
|
22 |
+
@patch("dataingestion.loaded_event_data", [])
|
23 |
+
def test_load_event_data_no_file_selected(
|
24 |
+
output_box_container, warning_box_container, warning_handler, mock_file_selector, filename_input, format_input, format_checkbox
|
25 |
+
):
|
26 |
+
# Set up file selector with no selection
|
27 |
+
mock_file_selector.value = []
|
28 |
+
load_event_data(
|
29 |
+
event=None,
|
30 |
+
file_selector=mock_file_selector,
|
31 |
+
filename_input=filename_input,
|
32 |
+
format_input=format_input,
|
33 |
+
format_checkbox=format_checkbox,
|
34 |
+
output_box_container=output_box_container,
|
35 |
+
warning_box_container=warning_box_container,
|
36 |
+
warning_handler=warning_handler,
|
37 |
+
)
|
38 |
+
assert "No file selected" in output_box_container[0].output_content
|
39 |
+
|
40 |
+
|
41 |
+
@patch("dataingestion.loaded_event_data", [])
|
42 |
+
@patch("dataingestion.EventList.read")
|
43 |
+
def test_load_event_data_success(mock_read, output_box_container, warning_box_container, warning_handler, mock_file_selector, filename_input, format_input, format_checkbox):
|
44 |
+
# Mock EventList read to return a valid event
|
45 |
+
mock_read.return_value = MagicMock()
|
46 |
+
|
47 |
+
load_event_data(
|
48 |
+
event=None,
|
49 |
+
file_selector=mock_file_selector,
|
50 |
+
filename_input=filename_input,
|
51 |
+
format_input=format_input,
|
52 |
+
format_checkbox=format_checkbox,
|
53 |
+
output_box_container=output_box_container,
|
54 |
+
warning_box_container=warning_box_container,
|
55 |
+
warning_handler=warning_handler,
|
56 |
+
)
|
57 |
+
assert len(output_box_container) > 0
|
58 |
+
assert "loaded successfully" in output_box_container[0].output_content
|
59 |
+
|
60 |
+
|
61 |
+
@patch("dataingestion.loaded_event_data", [("file1", MagicMock())])
|
62 |
+
def test_load_event_data_duplicate_file(
|
63 |
+
output_box_container, warning_box_container, warning_handler, mock_file_selector, filename_input, format_input, format_checkbox
|
64 |
+
):
|
65 |
+
# Test with duplicate file name
|
66 |
+
filename_input.value = "file1"
|
67 |
+
load_event_data(
|
68 |
+
event=None,
|
69 |
+
file_selector=mock_file_selector,
|
70 |
+
filename_input=filename_input,
|
71 |
+
format_input=format_input,
|
72 |
+
format_checkbox=format_checkbox,
|
73 |
+
output_box_container=output_box_container,
|
74 |
+
warning_box_container=warning_box_container,
|
75 |
+
warning_handler=warning_handler,
|
76 |
+
)
|
77 |
+
assert "already exists in memory" in output_box_container[0].output_content
|
78 |
+
|
79 |
+
|
80 |
+
@patch("dataingestion.os.path.exists", return_value=False)
|
81 |
+
@patch("dataingestion.loaded_event_data", [("file1", MagicMock())])
|
82 |
+
def test_save_loaded_files_success(mock_exists, output_box_container, warning_box_container, warning_handler, filename_input, format_input, format_checkbox):
|
83 |
+
save_loaded_files(
|
84 |
+
event=None,
|
85 |
+
filename_input=filename_input,
|
86 |
+
format_input=format_input,
|
87 |
+
format_checkbox=format_checkbox,
|
88 |
+
output_box_container=output_box_container,
|
89 |
+
warning_box_container=warning_box_container,
|
90 |
+
warning_handler=warning_handler,
|
91 |
+
)
|
92 |
+
assert "saved successfully" in output_box_container[0].output_content
|
93 |
+
|
94 |
+
|
95 |
+
@patch("dataingestion.os.path.exists", return_value=True)
|
96 |
+
@patch("dataingestion.loaded_event_data", [("file1", MagicMock())])
|
97 |
+
def test_save_loaded_files_duplicate_name(mock_exists, output_box_container, warning_box_container, warning_handler, filename_input, format_input, format_checkbox):
|
98 |
+
save_loaded_files(
|
99 |
+
event=None,
|
100 |
+
filename_input=filename_input,
|
101 |
+
format_input=format_input,
|
102 |
+
format_checkbox=format_checkbox,
|
103 |
+
output_box_container=output_box_container,
|
104 |
+
warning_box_container=warning_box_container,
|
105 |
+
warning_handler=warning_handler,
|
106 |
+
)
|
107 |
+
assert "already exists" in output_box_container[0].output_content
|
108 |
+
|
109 |
+
|
110 |
+
@patch("dataingestion.os.remove")
|
111 |
+
def test_delete_selected_files_success(mock_remove, output_box_container, warning_box_container, warning_handler, mock_file_selector):
|
112 |
+
delete_selected_files(
|
113 |
+
event=None,
|
114 |
+
file_selector=mock_file_selector,
|
115 |
+
output_box_container=output_box_container,
|
116 |
+
warning_box_container=warning_box_container,
|
117 |
+
warning_handler=warning_handler,
|
118 |
+
)
|
119 |
+
assert "deleted successfully" in output_box_container[0].output_content
|
120 |
+
|
121 |
+
|
122 |
+
def test_preview_loaded_files_no_data(output_box_container, warning_box_container, warning_handler):
|
123 |
+
preview_loaded_files(
|
124 |
+
event=None,
|
125 |
+
output_box_container=output_box_container,
|
126 |
+
warning_box_container=warning_box_container,
|
127 |
+
warning_handler=warning_handler,
|
128 |
+
)
|
129 |
+
assert "No valid files or light curves loaded" in output_box_container[0].output_content
|
130 |
+
|
131 |
+
|
132 |
+
@patch("dataingestion.loaded_event_data", [("event1", MagicMock(time=[0.1, 0.2], mjdref=58000, gti=[[0, 1]]) )])
|
133 |
+
def test_preview_loaded_files_with_data(output_box_container, warning_box_container, warning_handler):
|
134 |
+
preview_loaded_files(
|
135 |
+
event=None,
|
136 |
+
output_box_container=output_box_container,
|
137 |
+
warning_box_container=warning_box_container,
|
138 |
+
warning_handler=warning_handler,
|
139 |
+
)
|
140 |
+
assert "Event List - event1" in output_box_container[0].output_content
|
141 |
+
|
142 |
+
|
143 |
+
@patch("dataingestion.loaded_event_data", [("event1", MagicMock())])
|
144 |
+
def test_clear_loaded_files(output_box_container, warning_box_container):
|
145 |
+
clear_loaded_files(
|
146 |
+
event=None,
|
147 |
+
output_box_container=output_box_container,
|
148 |
+
warning_box_container=warning_box_container,
|
149 |
+
)
|
150 |
+
assert "cleared" in output_box_container[0].output_content
|
151 |
+
|
152 |
+
|
153 |
+
def test_create_event_list_missing_data(output_box_container, warning_box_container, warning_handler):
|
154 |
+
create_event_list(
|
155 |
+
event=None,
|
156 |
+
times_input=MagicMock(value=""),
|
157 |
+
energy_input=MagicMock(value=""),
|
158 |
+
pi_input=MagicMock(value=""),
|
159 |
+
gti_input=MagicMock(value=""),
|
160 |
+
mjdref_input=MagicMock(value=""),
|
161 |
+
name_input=MagicMock(value=""),
|
162 |
+
output_box_container=output_box_container,
|
163 |
+
warning_box_container=warning_box_container,
|
164 |
+
warning_handler=warning_handler,
|
165 |
+
)
|
166 |
+
assert "Please enter Photon Arrival Times and MJDREF" in output_box_container[0].output_content
|
167 |
+
|
168 |
+
|
169 |
+
def test_simulate_event_list(output_box_container, warning_box_container, warning_handler):
|
170 |
+
simulate_event_list(
|
171 |
+
event=None,
|
172 |
+
time_slider=MagicMock(value=10),
|
173 |
+
count_slider=MagicMock(value=5),
|
174 |
+
dt_input=MagicMock(value=0.1),
|
175 |
+
name_input=MagicMock(value="simulated_event"),
|
176 |
+
method_selector=MagicMock(value="Standard Method"),
|
177 |
+
output_box_container=output_box_container,
|
178 |
+
warning_box_container=warning_box_container,
|
179 |
+
warning_handler=warning_handler,
|
180 |
+
)
|
181 |
+
assert "simulated successfully" in output_box_container[0].output_content
|
182 |
+
|
183 |
+
|
184 |
+
def test_create_warning_handler():
|
185 |
+
handler = create_warning_handler()
|
186 |
+
with pytest.warns(None) as record:
|
187 |
+
handler.warn("Test warning", category=UserWarning)
|
188 |
+
assert len(record) == 1
|
189 |
+
assert record[0].message.args[0] == "Test warning"
|