1.增加Unittest

2.考虑利用实际地闪密度和雷电流曲线。
3.版本号v1.2.0

untest.py

@@ -0,0 +1,29 @@
+import unittest
+from core import thunder_density
+from main import read_parameter
+import numpy as np
+import sympy
+
+
+class Testing(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        # read_parameter('default.toml')
+        pass
+
+    def test_thunder_density(self):
+        i = sympy.symbols("i")
+        a = np.random.random()
+        b = np.random.random()
+        p = 1-1 / (1 + (i / a) ** b)
+        d_p = sympy.diff(p, i)
+        random_i = np.random.randint(10, 100)
+        v_from_thunder_density = thunder_density(random_i, 0, a, b)
+        v_from_diff = d_p.evalf(subs={i: random_i})
+        self.assertTrue(
+            abs(v_from_thunder_density - v_from_diff) < 1e-5, "与自动微分结果不一致"
+        )  # add assertion here
+
+
+if __name__ == "__main__":
+    unittest.main()