added test for the beam_profile function.

ARM-DOE · Jan 19, 2024 · 59d2777 · 59d2777
1 parent 62b0ea9
commit 59d2777
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_gfit.py b/tests/test_gfit.py
@@ -34,4 +34,7 @@ def test_psds():
     np.testing.assert_almost_equal(my_psds['ScatMassEnsemble'].sum(), 3.15026266)
     np.testing.assert_almost_equal(my_psds['IncanMassEnsemble'].sum(), 0.08177226)
     np.testing.assert_almost_equal(my_binary['DeadtimeRelativeBias'].mean(), -0.00023515)
-    coeff, beam_profile = pysp2.util.beam_shape(my_binary,beam_position_from='peak maximum',Globals=pysp2.util.DMTGlobals())
+    coeff, beam_profile = pysp2.util.beam_shape(
+        my_binary, beam_position_from='peak maximum', Globals=pysp2.util.DMTGlobals())
+    np.testing.assert_almost_equal(coeff, [9.83851858e-01, 4.64317390e+01, 
+                                           1.14337852e+01, 4.46761788e-03])