From c906ba82f3772173a7e90bd0918c846530439a8b Mon Sep 17 00:00:00 2001 From: Jie Han Date: Tue, 19 May 2020 18:56:16 +0800 Subject: [PATCH] refactor: move import pytest line of blockchain algs under "main" section. (#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 --- blockchain/diophantine_equation.py | 5 ++--- blockchain/modular_division.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/blockchain/diophantine_equation.py b/blockchain/diophantine_equation.py index ec2ed26e4..dab4b3a65 100644 --- 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 index 1255f0432..c81c2138d 100644 --- 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)