UP my solution

Author: Ahmed-Semah

numpy_questions.py

@@ -48,7 +48,6 @@ def max_index(X):
     return i, j
 
 
-
 def wallis_product(n_terms):
     """Implement the Wallis product to compute an approximation of pi.
 
@@ -72,7 +71,7 @@ def wallis_product(n_terms):
     if n_terms < 0:
         raise ValueError("n_terms should be a positive integer")
     if n_terms == 0:
-        return 1.
+        return 2.
 
     pi = 1.
     for i in range(1, n_terms+1):

test_numpy_questions.py

@@ -9,6 +9,7 @@
 
 from numpy_questions import wallis_product, max_index
 
+
 def test_max_index():
     X = np.array([[0, 1], [2, 0]])
     assert max_index(X) == (1, 0)
@@ -29,7 +30,7 @@ def test_max_index():
 
 def test_wallis_product():
     pi_approx = wallis_product(0)
-    assert pi_approx == 1.
+    assert pi_approx == 2.
 
     pi_approx = wallis_product(1)
     assert pi_approx == 8 / 3