From cf44fe495295cde72eeb1f56e983e75955327a74 Mon Sep 17 00:00:00 2001
From: jiangyi15 <jiangyi15@mails.ucas.ac.cn>
Date: Tue, 1 Aug 2023 17:28:07 +0800
Subject: [PATCH] ci: add tests for lazycall

---
 tf_pwa/tests/config_lazycall.yml | 49 ++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 tf_pwa/tests/config_lazycall.yml

diff --git a/tf_pwa/tests/config_lazycall.yml b/tf_pwa/tests/config_lazycall.yml
new file mode 100644
index 00000000..735a889d
--- /dev/null
+++ b/tf_pwa/tests/config_lazycall.yml
@@ -0,0 +1,49 @@
+data:
+  dat_order: [B, C, D]
+  data: ["toy_data/data.dat"]
+  bg: ["toy_data/bg.dat"]
+  phsp: ["toy_data/PHSP.dat"]
+  random_z: False
+  r_boost: False
+  bg_weight: 0.1
+  lazy_call: True
+  use_tf_function: True
+  no_id_cached: True
+  jit_compile: True
+  cached_lazy_call: toy_data/cached/
+
+decay:
+  A:
+    - [R_BC, D]
+    - [R_BD, C]
+    - [R_CD, B]
+  R_BC: [B, C]
+  R_BD: [B, D]
+  R_CD: [C, D]
+
+particle:
+  $top:
+    A: { J: 1, P: -1, spins: [-1, 1], mass: 4.6 }
+  $finals:
+    B: { J: 1, P: -1, mass: 2.00698 }
+    C: { J: 1, P: -1, mass: 2.01028 }
+    D: { J: 0, P: -1, mass: 0.13957 }
+  R_BC: { J: 1, Par: 1, m0: 4.16, g0: 0.1, params: { mass_range: [4.0, 4.2] } }
+  R_BD: { J: 1, Par: 1, m0: 2.43, g0: 0.3 }
+  R_CD: { J: 1, Par: 1, m0: 2.42, g0: 0.03 }
+
+constrains:
+  particle: null
+  decay: null
+
+plot:
+  mass:
+    R_BC: { display: "$M_{BC}$" }
+    R_BD: { display: "$M_{BD}$" }
+    R_CD: { display: "$M_{CD}$" }
+  angle:
+    R_BC/B:
+      cos(beta):
+        display: "cos \\theta"
+      alpha:
+        display: "\\phi"