Created the main function

diff --git a/src/add/fsqrt.py b/src/add/fsqrt.py
index 948ad1cf483adb11edacfeae8ba96262f5b30743..c76267e7cc6b900d8cff941235dba02b37b1b9df 100644 (file)
--- a/src/add/fsqrt.py
+++ b/src/add/fsqrt.py
@@ -44,10 +44,18 @@ def sqrt(num):
         r = R
     return Q
 
-
-for Q in range(1, 26):
-    print(sqrt(Q), sqrtsimple(Q))
-
+def main(mantissa, exponent):
+    if exponent & 1 != 0:
+        return Q(sqrt(mantissa << 1), # shift mantissa up
+                ((exponent - 1) / 2)) # subtract 1 from exp to compensate
+    else:
+        return Q(sqrt(mantissa),      # mantissa as-is
+                (exponent / 2))       # no compensating needed on exp
+
+for Q in range(1, int(1e7)):
+    print(Q, sqrt(Q), sqrtsimple(Q), int(Q**0.5))
+    assert int(Q**0.5) == sqrtsimple(Q), "Q sqrtsimpl fail %d" % Q
+    assert int(Q**0.5) == sqrt(Q), "Q sqrt fail %d" % Q
 """
 //This is the main code of integer sqrt function found here:http://verilogcodes.blogspot.com/2017/11/a-verilog-function-for-finding-square-root.html
 //