From c4170c5e724ad5869ba772148eddfcb8efd23697 Mon Sep 17 00:00:00 2001
From: Jean-Baptiste Bayle <j2b.bayle@gmail.com>
Date: Wed, 24 Aug 2022 16:35:26 +0200
Subject: [PATCH] Add tests for supported glitch files
---
.gitattributes | 2 +
tests/glitch-1-0.h5 | 3 +
tests/glitch-1-1.h5 | 3 +
tests/test_glitch.py | 178 +++++++++++++++++++++++++++++++++++++++++++
4 files changed, 186 insertions(+)
create mode 100644 tests/glitch-1-0.h5
create mode 100644 tests/glitch-1-1.h5
create mode 100644 tests/test_glitch.py
diff --git a/.gitattributes b/.gitattributes
index df50dbe..82bfcf8 100755
--- a/.gitattributes
+++ b/.gitattributes
@@ -6,3 +6,5 @@ tests/keplerian-fplan-1-1.h5 filter=lfs diff=lfs merge=lfs -text
tests/esa-trailing-fplan-1-1.h5 filter=lfs diff=lfs merge=lfs -text
tests/gws-1-1.h5 filter=lfs diff=lfs merge=lfs -text
tests/gws-2-0-dev.h5 filter=lfs diff=lfs merge=lfs -text
+tests/glitch-1-1.h5 filter=lfs diff=lfs merge=lfs -text
+tests/glitch-1-0.h5 filter=lfs diff=lfs merge=lfs -text
diff --git a/tests/glitch-1-0.h5 b/tests/glitch-1-0.h5
new file mode 100644
index 0000000..c5c0bc4
--- /dev/null
+++ b/tests/glitch-1-0.h5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cf82436640f19ce1716fb97c67a2cedd64dbd5db97867376604c705417579c4
+size 45552
diff --git a/tests/glitch-1-1.h5 b/tests/glitch-1-1.h5
new file mode 100644
index 0000000..c5c0bc4
--- /dev/null
+++ b/tests/glitch-1-1.h5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cf82436640f19ce1716fb97c67a2cedd64dbd5db97867376604c705417579c4
+size 45552
diff --git a/tests/test_glitch.py b/tests/test_glitch.py
new file mode 100644
index 0000000..762744a
--- /dev/null
+++ b/tests/test_glitch.py
@@ -0,0 +1,178 @@
+#! /usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""Test reading of glitch files."""
+
+import numpy as np
+from pytest import approx
+from h5py import File
+from lisaconstants import c
+from lisainstrument import Instrument
+
+
+def test_no_glitches():
+ """Test that we can simulate no glitches."""
+
+ instru = Instrument(size=100, glitches=None)
+ instru.disable_all_noises()
+
+ assert instru.glitch_file is None
+ for mosa in instru.MOSAS:
+ assert instru.glitch_tms[mosa] == 0
+ assert instru.glitch_lasers[mosa] == 0
+
+ instru.simulate()
+
+ for mosa in instru.MOSAS:
+ assert instru.local_carrier_fluctuations[mosa] == approx(0)
+ assert instru.local_tmi_carrier_fluctuations[mosa] == approx(0)
+ assert instru.local_tmi_usb_fluctuations[mosa] == approx(0)
+
+def test_glitch_file_1_0():
+ """Test that we can read test-mass and laser glitches from files v1.0.
+
+ Test glitch file can be generated using LISA Glitch and the following script.
+
+ from lisaglitch import ShapeletGlitch
+ path = 'tests/glitch-1-0.h5'
+ ShapeletGlitch(beta=2, quantum_n=1, inj_point='tm_12', t_inj=10.0, size=200).write(path, mode='w')
+ ShapeletGlitch(beta=1, quantum_n=2, inj_point='tm_23', t_inj=0.0).write(path)
+ ShapeletGlitch(beta=3, quantum_n=2, inj_point='tm_31', t_inj=5.0).write(path)
+ ShapeletGlitch(beta=4, quantum_n=1, inj_point='laser_12', t_inj=0.0).write(path)
+ ShapeletGlitch(beta=4, quantum_n=3, inj_point='laser_31', t_inj=0.0).write(path)
+ """
+
+ glitch_file = 'tests/glitch-1-0.h5'
+ glitchf = File(glitch_file, 'r')
+
+ instru = Instrument(
+ size=glitchf.attrs['size'],
+ t0=glitchf.attrs['t0'],
+ dt=glitchf.attrs['dt'],
+ physics_upsampling=1,
+ aafilter=None,
+ lock='six',
+ glitches=glitch_file,
+ orbits='tests/keplerian-orbits-1-0-2.h5')
+ instru.disable_all_noises()
+
+ assert instru.glitch_file is glitch_file
+ for mosa in instru.MOSAS:
+
+ # Missing datasets should be zero
+ tm_dataset = f'tm_{mosa}'
+ if tm_dataset in glitchf:
+ assert instru.glitch_tms[mosa] == approx(glitchf[tm_dataset][:])
+ else:
+ assert instru.glitch_tms[mosa] == 0
+
+ laser_dataset = f'laser_{mosa}'
+ if laser_dataset in glitchf:
+ assert instru.glitch_lasers[mosa] == approx(glitchf[laser_dataset][:])
+ else:
+ assert instru.glitch_lasers[mosa] == 0
+
+ instru.simulate()
+
+ for mosa in instru.MOSAS:
+
+ laser_dataset = f'laser_{mosa}'
+ if laser_dataset in glitchf:
+ assert approx(instru.local_carrier_fluctuations[mosa]) == \
+ glitchf[laser_dataset][:]
+ else:
+ assert approx(instru.local_carrier_fluctuations[mosa]) == 0
+
+ tm_dataset = f'tm_{mosa}'
+
+ if tm_dataset in glitchf:
+ assert approx(instru.local_tmi_carrier_fluctuations[mosa], abs=1E-6) == \
+ instru.local_carrier_fluctuations[mosa] \
+ + 2 * glitchf[tm_dataset][:] / c \
+ * (instru.central_freq + instru.local_tmi_carrier_offsets[mosa])
+ else:
+ assert approx(instru.local_tmi_carrier_fluctuations[mosa]) == \
+ instru.local_carrier_fluctuations[mosa]
+
+ if tm_dataset in glitchf:
+ assert approx(instru.local_tmi_usb_fluctuations[mosa], abs=1E-6) == \
+ instru.local_usb_fluctuations[mosa] \
+ + 2 * glitchf[tm_dataset][:] / c \
+ * (instru.central_freq + instru.local_tmi_usb_offsets[mosa])
+ else:
+ assert approx(instru.local_tmi_usb_fluctuations[mosa]) == \
+ instru.local_usb_fluctuations[mosa]
+
+def test_glitch_file_1_1():
+ """Test that we can read test-mass and laser glitches from files v1.1.
+
+ Test glitch file can be generated using LISA Glitch and the following script.
+
+ from lisaglitch import ShapeletGlitch
+ path = 'tests/glitch-1-1.h5'
+ ShapeletGlitch(beta=2, quantum_n=1, inj_point='tm_12', t_inj=10.0, size=200).write(path, mode='w')
+ ShapeletGlitch(beta=1, quantum_n=2, inj_point='tm_23', t_inj=0.0).write(path)
+ ShapeletGlitch(beta=3, quantum_n=2, inj_point='tm_31', t_inj=5.0).write(path)
+ ShapeletGlitch(beta=4, quantum_n=1, inj_point='laser_12', t_inj=0.0).write(path)
+ ShapeletGlitch(beta=4, quantum_n=3, inj_point='laser_31', t_inj=0.0).write(path)
+ """
+
+ glitch_file = 'tests/glitch-1-1.h5'
+ glitchf = File(glitch_file, 'r')
+
+ instru = Instrument(
+ size=glitchf.attrs['size'],
+ t0=glitchf.attrs['t0'],
+ dt=glitchf.attrs['dt'],
+ physics_upsampling=1,
+ aafilter=None,
+ lock='six',
+ glitches=glitch_file,
+ orbits='tests/keplerian-orbits-1-0-2.h5')
+ instru.disable_all_noises()
+
+ assert instru.glitch_file is glitch_file
+ for mosa in instru.MOSAS:
+
+ # Missing datasets should be zero
+ tm_dataset = f'tm_{mosa}'
+ if tm_dataset in glitchf:
+ assert instru.glitch_tms[mosa] == approx(glitchf[tm_dataset][:])
+ else:
+ assert instru.glitch_tms[mosa] == 0
+
+ laser_dataset = f'laser_{mosa}'
+ if laser_dataset in glitchf:
+ assert instru.glitch_lasers[mosa] == approx(glitchf[laser_dataset][:])
+ else:
+ assert instru.glitch_lasers[mosa] == 0
+
+ instru.simulate()
+
+ for mosa in instru.MOSAS:
+
+ laser_dataset = f'laser_{mosa}'
+ if laser_dataset in glitchf:
+ assert approx(instru.local_carrier_fluctuations[mosa]) == \
+ glitchf[laser_dataset][:]
+ else:
+ assert approx(instru.local_carrier_fluctuations[mosa]) == 0
+
+ tm_dataset = f'tm_{mosa}'
+
+ if tm_dataset in glitchf:
+ assert approx(instru.local_tmi_carrier_fluctuations[mosa], abs=1E-6) == \
+ instru.local_carrier_fluctuations[mosa] \
+ + 2 * glitchf[tm_dataset][:] / c \
+ * (instru.central_freq + instru.local_tmi_carrier_offsets[mosa])
+ else:
+ assert approx(instru.local_tmi_carrier_fluctuations[mosa]) == \
+ instru.local_carrier_fluctuations[mosa]
+
+ if tm_dataset in glitchf:
+ assert approx(instru.local_tmi_usb_fluctuations[mosa], abs=1E-6) == \
+ instru.local_usb_fluctuations[mosa] \
+ + 2 * glitchf[tm_dataset][:] / c \
+ * (instru.central_freq + instru.local_tmi_usb_offsets[mosa])
+ else:
+ assert approx(instru.local_tmi_usb_fluctuations[mosa]) == \
+ instru.local_usb_fluctuations[mosa]
--
GitLab