From a2ec0df56181893f4623942a1a5ef98557a36ddc Mon Sep 17 00:00:00 2001 From: Luke Kenneth Casson Leighton Date: Sun, 28 Apr 2019 14:44:18 +0100 Subject: [PATCH] add an __main__ test thing --- src/add/fsqrt.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/src/add/fsqrt.py b/src/add/fsqrt.py index cb6a8ac0..3a8f8905 100644 --- a/src/add/fsqrt.py +++ b/src/add/fsqrt.py @@ -44,6 +44,7 @@ def sqrt(num): r = R return Q + def main(mantissa, exponent): if exponent & 1 != 0: return sqrt(mantissa << 1), # shift mantissa up @@ -51,10 +52,13 @@ def main(mantissa, exponent): return 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 + +if __name__ == '__main__': + 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 // -- 2.30.2