From a745ec647cf1107f5af69cf999a7025fd67aaaa8 Mon Sep 17 00:00:00 2001 From: Pablito Date: Sat, 7 Dec 2024 14:12:19 +0100 Subject: [PATCH] Added tests to proth_number.py --- maths/special_numbers/proth_number.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/maths/special_numbers/proth_number.py b/maths/special_numbers/proth_number.py index 47747ed26..fa2a4b69b 100644 --- a/maths/special_numbers/proth_number.py +++ b/maths/special_numbers/proth_number.py @@ -59,6 +59,16 @@ def proth(number: int) -> int: return proth_list[number - 1] +def test_proth_valid() -> None: + """Test the 6 first Proth numbers.""" + assert proth(1) == 3, "The 1st Proth number should be 3" + assert proth(2) == 5, "The 2nd Proth number should be 5" + assert proth(3) == 9, "The 3rd Proth number should be 9" + assert proth(4) == 13, "The 4th Proth number should be 13" + assert proth(5) == 17, "The 5th Proth number should be 17" + assert proth(6) == 25, "The 6th Proth number should be 25" + + if __name__ == "__main__": import doctest