From 7f7b1f39f30dbc5c9247530916898a479711750b Mon Sep 17 00:00:00 2001 From: BamaCharanChhandogi Date: Tue, 3 Oct 2023 00:46:26 +0530 Subject: [PATCH] add doctest in main method for testing --- maths/combinations.py | 19 +++---------------- 1 file changed, 3 insertions(+), 16 deletions(-) diff --git a/maths/combinations.py b/maths/combinations.py index a2324012c01f..ef3a018a729a 100644 --- a/maths/combinations.py +++ b/maths/combinations.py @@ -39,19 +39,6 @@ def combinations(n: int, k: int) -> int: if __name__ == "__main__": - print( - "The number of five-card hands possible from a standard", - f"fifty-two card deck is: {combinations(52, 5)}\n", - ) - - print( - "If a class of 40 students must be arranged into groups of", - f"4 for group projects, there are {combinations(40, 4)} ways", - "to arrange them.\n", - ) - - print( - "If 10 teams are competing in a Formula One race, there", - f"are {combinations(10, 3)} ways that first, second and", - "third place can be awarded.", - ) + import doctest; + + doctest.testmod();