some pytest on math folder

maths/3n+1.py

@@ -32,6 +32,14 @@ def main():
         )
     )
 
+def testn31():
+    assert n31(4) == ([4, 2, 1], 3)
+
+    assert n31(11) == ([11, 34, 17, 52, 26, 13, 40, 20, 10, 5, 16, 8, 4, 2, 1], 15)
+
+    assert n31(31) == ([31, 94, 47, 142, 71, 214, 107, 322, 161, 484, 242, 121, 364, 182, 91, 274, 137, 412, 206, 103, 310, 155, 466, 233, 700, 350, 175, 526, 263, 790, 395, 1186, 593, 1780, 890, 445, 1336, 668, 334, 167, 502, 251, 754, 377, 1132, 566, 283, 850, 425, 1276, 638, 319, 958, 479, 1438, 719, 2158, 1079, 3238, 1619, 4858, 2429, 7288, 3644, 1822, 911, 2734, 1367, 4102, 2051, 6154, 3077, 9232, 4616, 2308, 1154, 577, 1732, 866, 433, 1300, 650, 325, 976, 488, 244, 122, 61, 184, 92, 46, 23, 70, 35, 106, 53, 160, 80, 40, 20, 10, 5, 16, 8, 4, 2, 1], 107)
 
 if __name__ == "__main__":
     main()
+    testn31()
+

maths/abs.py

@@ -18,5 +18,13 @@ def main():
     print(abs_val(-34))  # = 34
 
 
+def testAbs():
+    assert 0 == abs_val(0)
+
+    assert 34 == abs_val(34)
+
+    assert 100000000000 == abs_val(-100000000000)
+
 if __name__ == "__main__":
     main()
+    testAbs()

maths/average_mean.py

@@ -12,5 +12,13 @@ def main():
     print(average([2, 4, 6, 8, 20, 50, 70]))
 
 
+def testAverage():
+    assert 12.0 == average([3, 6, 9, 12, 15, 18, 21])
+
+    assert 20 == average([5, 10, 15, 20, 25, 30, 35])
+
+    assert 4.5 == average([1, 2, 3, 4, 5, 6, 7, 8])
+
 if __name__ == "__main__":
     main()
+    testAverage()