refactor: move import pytest line of blockchain algs under "main" sec…

…tion. (TheAlgorithms#2012) * change doctest line import doctest is not relevant with algorithms. move it under main section. * from doctest import testmod * refactor: move doctest under "main" section * Update diophantine_equation.py * Update modular_division.py Co-authored-by: Christian Clauss <[email protected]>
Woundrite · May 19, 2020 · c906ba8 · c906ba8
1 parent 1c62bd1
commit c906ba8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/blockchain/diophantine_equation.py b/blockchain/diophantine_equation.py
@@ -118,10 +118,9 @@ def extended_gcd(a, b):
     return (d, x, y)
 
 
-# import testmod for testing our function
-from doctest import testmod
-
 if __name__ == "__main__":
+    from doctest import testmod
+
     testmod(name="diophantine", verbose=True)
     testmod(name="diophantine_all_soln", verbose=True)
     testmod(name="extended_gcd", verbose=True)

diff --git a/blockchain/modular_division.py b/blockchain/modular_division.py
@@ -139,10 +139,9 @@ def greatest_common_divisor(a, b):
     return b
 
 
-# Import testmod for testing our function
-from doctest import testmod
-
 if __name__ == "__main__":
+    from doctest import testmod
+
     testmod(name="modular_division", verbose=True)
     testmod(name="modular_division2", verbose=True)
     testmod(name="invert_modulo", verbose=True)