Merge pull request #344 from lucafedeli88/qed_phys_part_with_lambda

QED: now particles initialize a real component (optical depth) to be used for QED modules

1 files changed, 36 insertions, 0 deletions

diff --git a/Examples/Modules/qed/quantum_synchrotron/check_2d_tau_init.py b/Examples/Modules/qed/quantum_synchrotron/check_2d_tau_init.py
new file mode 100755
index 000000000..05b313ee6
--- /dev/null
+++ b/Examples/Modules/qed/quantum_synchrotron/check_2d_tau_init.py
@@ -0,0 +1,36 @@
+#! /usr/bin/env python3
+import yt
+import numpy as np
+import scipy.stats as st
+import sys
+
+# This script checks if electrons and positrons initialized with
+# Quantum Synchrotron process enabled
+# do actually have an exponentially distributed optical depth
+
+# Tolerance
+tol = 1e-2
+
+def check():
+    filename = sys.argv[1]
+    data_set = yt.load(filename)
+
+    all_data = data_set.all_data()
+    res_ele_tau = all_data["electrons", 'particle_tau']
+    res_pos_tau = all_data["positrons", 'particle_tau']
+
+    loc_ele, scale_ele = st.expon.fit(res_ele_tau)
+    loc_pos, scale_pos = st.expon.fit(res_pos_tau)
+
+    # loc should be very close to 0, scale should be very close to 1
+    assert(np.abs(loc_ele - 0) < tol)
+    assert(np.abs(loc_pos - 0) < tol)
+    assert(np.abs(scale_ele - 1) < tol)
+    assert(np.abs(scale_pos - 1) < tol)
+
+def main():
+    check()
+
+if __name__ == "__main__":
+    main()
+