updated log

calc/__init__.py

@@ -1,4 +1,5 @@
 import numpy as np
+import math
 
 
 def add(a, b):
@@ -21,4 +22,11 @@ def pow(a, b):
     return a*b
 
 def log(a, base=10):
-    return np.log(a)
+	"""
+	Return the log of a, base 10 (by default).
+
+	Parameters:
+	-----------
+	a: int
+	"""
+	return math.log(a, base)

calc/tests/test_calc.py

@@ -1,5 +1,8 @@
 from calc import add
 
-def test_add():
-    assert add(1, 2) == 3
+def test_log():
+	assert log(100) == 2
+	assert log(8, 2) == 3
 
+def test_add():
+    assert add(1, 2) == 3