Merge pull request #10 from edwardoughton/enhance_tests

Enhance tests

src/itmlogic/qerf.py

@@ -1,20 +1,22 @@
+import math
+
 def qerf(z):
     """
-    The standard normal complementary probability - see function in 
+    The standard normal complementary probability - see function in
     C. Hastings, Jr. (1955). The maximum error should be 7.5x10^-8.
 
     Parameters
     ----------
     z : TODO: Type
         TODO: Description
-    
+
     Output
     ------
     qerf1 : float
         TODO:
 
     """
-    
+
     b1 = 0.319381530
     b2 = -0.356563782
     b3 = 1.781477937
@@ -24,22 +26,23 @@ def qerf(z):
 
     rrt2pi = 0.398942280
 
-    x = z[0]
+    x = z
     t = abs(x)
 
     if t >= 10:
-    
+
         qerf1 = 0
-    
+
     else:
-    
+
         t = rp / (t + rp)
 
         qerf1 = (
-            exp(-0.5 * x**2) * rrt2pi * 
+            math.exp(-0.5 * x**2) * rrt2pi *
             ((((b5 * t + b4) * t + b3) * t + b2) * t + b1) * t
             )
 
-    # if (x<0.) qerf1=1.-qerf1  TODO:
+    if x < 0:
+        qerf1 = 1 - qerf1
 
-    return qerf1
+    return qerf1

tests/test_aknfe.py

@@ -3,8 +3,7 @@
 
 def test_aknfe():
 
+    assert round(aknfe(2), 2) == 16.36
+    assert round(aknfe(3), 2) == 17.99
     assert round(aknfe(5), 2) == 20.04
     assert round(aknfe(6), 2) == 20.73
-
-
-

tests/test_fht.py

@@ -12,5 +12,6 @@ def test_fht():
 
     assert actual_answer == expected_answer
 
+    assert round(fht(150, 20), 2) == 11.05
 
-
+    assert round(fht(150, 1e-6), 2) == -29.96

tests/test_h0f.py

@@ -12,3 +12,11 @@ def test_h0f():
     expected_answer = 34.28154717133048
 
     assert actual_answer == expected_answer
+
+    actual_answer = h0f(2, -1)
+
+    assert round(actual_answer, 2) == 9.33
+
+    actual_answer = h0f(2, 6)
+
+    assert round(actual_answer, 2) == 18.53

tests/test_qerf.py

@@ -1,4 +1,8 @@
 import pytest
 from itmlogic.qerf import qerf
 
-# def test_qerf():
+def test_qerf():
+
+    assert round(qerf(-1),4) == 0.8413
+    assert round(qerf(1),4) == 0.1587
+    assert round(qerf(10),4) == 0