From 3cb93f68164f32594bfdce1cb829564132e39e35 Mon Sep 17 00:00:00 2001
From: Philipp Niedermayer <p.niedermayer@gsi.de>
Date: Wed, 4 May 2022 16:07:42 +0200
Subject: [PATCH] Add test
---
.gitignore | 10 ++++++++++
test/test.py | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 61 insertions(+)
create mode 100644 test/test.py
diff --git a/.gitignore b/.gitignore
index 05933a1..078ab50 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,14 @@
+# Data files
+*.bin
+*.tdf
+
+
+
+
+
+
+
# Created by https://www.toptal.com/developers/gitignore/api/python,jupyternotebooks
# Edit at https://www.toptal.com/developers/gitignore?templates=python,jupyternotebooks
diff --git a/test/test.py b/test/test.py
new file mode 100644
index 0000000..8f57471
--- /dev/null
+++ b/test/test.py
@@ -0,0 +1,51 @@
+
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
+from common.utils import *
+
+
+import matplotlib as mpl
+from matplotlib import pyplot as plt
+
+# https://arxiv.org/abs/2107.02270
+CB_color_cycle =["#3f90da", "#ffa90e", "#bd1f01", "#94a4a2", "#832db6", "#a96b59", "#e76300", "#b9ac70", "#717581", "#92dadd"]
+cmap = mpl.colors.ListedColormap(CB_color_cycle, 'Petroff10')
+mpl.rcParams['axes.prop_cycle'] = mpl.cycler(color=CB_color_cycle)
+
+
+
+fig, ax = plt.subplots(5,1, sharex='col', figsize=(5,8))
+
+data = LassieSpillData.from_file('test_2022-05-04_13 16 50_asl340.acc.gsi.de_GS01DT_ML_pniederm_test_42c_00014.tdf',
+ from_event=CMD_BEAM_INJECTION)
+ax[0].plot(data.t, data.v)
+ax[0].set(title='BCT', ylabel=data.unit)
+
+
+data = LiberaBBBData.from_file('test_20220504-151238_sdlibera223_bpm2_bbb_.bin')
+ax[1].plot(data.t, data.x, label=f'x / {data.x_unit}')
+ax[1].plot(data.t, data.y, label=f'y / {data.y_unit}')
+ax[1].legend()
+ax[1].set(title='BPM', ylabel=data.x_unit)
+ax[2].plot(data.t, data.s)
+ax[2].set(title='BPM sum', ylabel=data.s_unit)
+ax[3].plot(*avg(data.t, data.f, n=1000))
+ax[3].set(title='BPM bunch frequency', ylabel=data.f_unit)
+
+
+ax[-1].set(xlabel='time / s')
+plt.show()
+
+
+
+
+
+
+
+
+
+
+
+
+
--
GitLab