Skip to content
Snippets Groups Projects
Commit df6a16cd authored by Kevin Kuns's avatar Kevin Kuns
Browse files

adjustments after refactoring

parent e673aa0f
No related branches found
No related tags found
1 merge request!155Budget tests
Pipeline #482957 passed
......@@ -153,8 +153,8 @@ is available in the `budget.ifo` attribute.
The budget `run()` method calculates all budget noises and the noise
total and returns a `BudgetTrace` object with `freq`, `psd`, and `asd`
properties. The budget sub-traces are available through a dictionary
(`trace['QuantumVacuum']`) interface and via attributes
(`trace.QuantumVacumm`).
(`trace['Quantum']`) interface and via attributes
(`trace.Quantum`).
The budget `freq` and `ifo` attributes can be updated at run time by
passing them as keyword arguments to the `run()` method:
......@@ -438,13 +438,13 @@ and extract the noise data the output traces dictionary:
```python
budget = load_budget('/path/to/MyBudget', freq)
trace = budget.run()
quantum_trace = trace['QuantumVacuum']
quantum_trace = trace['Quantum']
```
You can also calculate the final calibrated output noise for just a
single term using the Budget `calc_noise()` method:
```python
data = budget.calc_noise('QuantumVacuum')
data = budget.calc_noise('Quantum')
```
You can also calculate a noise at it's source, without applying any
......@@ -452,7 +452,7 @@ calibrations, by using the Budget `__getitem__` interface to extract
the specific Noise BudgetItem for the noise you're interested in, and
running it's `calc_trace()` method directly:
```python
data = budget['QuantumVacuum'].calc_trace()
data = budget['Quantum'].calc_trace()
```
......
......@@ -65,12 +65,12 @@ def test_budget_run_calc(tpath_join, pprint, compare_noise):
F_Hz = np.logspace(np.log10(5), 4, 3000)
B = load_budget('CE2silica', freq=F_Hz)
traces1 = B.run()
traces2 = B.calc_noise('QuantumVacuum')
traces3 = B['QuantumVacuum'].calc_trace()
traces4 = B['QuantumVacuum'].calc_noise('QuantumVacuumAS')
traces5 = B['QuantumVacuum']['QuantumVacuumAS'].calc_trace()
traces6 = B['QuantumVacuum'].run()
fig1 = traces1.QuantumVacuum.plot()
traces2 = B.calc_noise('Quantum')
traces3 = B['Quantum'].calc_trace()
traces4 = B['Quantum'].calc_noise('AS')
traces5 = B['Quantum']['AS'].calc_trace()
traces6 = B['Quantum'].run()
fig1 = traces1.Quantum.plot()
fig2 = traces2.plot()
fig3 = traces3.plot()
fig1.savefig(tpath_join('run.pdf'))
......@@ -78,8 +78,8 @@ def test_budget_run_calc(tpath_join, pprint, compare_noise):
fig3.savefig(tpath_join('calc_trace.pdf'))
pprint('Testing that run() and calc_noise() do the same thing')
compare_noise(traces1.QuantumVacuum, traces2)
compare_noise(traces1.QuantumVacuum.QuantumVacuumAS, traces2.QuantumVacuumAS)
compare_noise(traces1.Quantum, traces2)
compare_noise(traces1.Quantum.AS, traces2.AS)
pprint(
"Testing that run() and calc_trace() on sub-budgets don't apply strain calibration")
compare_noise(traces3, traces6)
......@@ -96,7 +96,7 @@ def test_budget_run_calc(tpath_join, pprint, compare_noise):
update_total_psd(traces3)
fig, ax = plt.subplots()
ax.loglog(F_Hz, traces1.QuantumVacuum.asd, label='run')
ax.loglog(F_Hz, traces1.Quantum.asd, label='run')
ax.loglog(F_Hz, traces2.asd, ls='--', label='calc_noise')
ax.loglog(F_Hz, traces3.asd, ls='-.', label='calc_trace / $L_\mathrm{arm}$')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment