Sample code for the doctest article

Author: lpozo

python-doctest/README.md

@@ -0,0 +1,3 @@
+# Python's doctest: Document and Test You Code at Once
+
+This folder provides the code examples for the article [Python's doctest: Document and Test You Code at Once](https://realpython.com/python-doctest/).

python-doctest/broken_tests.txt

@@ -0,0 +1,20 @@
+>>> # The output must be an integer
+>>> 5 + 7
+12.0
+
+>>> # The output must not contain quotes
+>>> print("Hello, World!")
+'Hello, World!'
+
+>>> # The output must not use double quotes
+>>> "Hello," + "World!"
+"Hello, World!"
+
+>>> # The output must not contain leading or trailing spaces
+>>> print("Hello, World!")
+  Hello, World!
+
+>>> # The traceback doesn't include the correct exception message
+>>> raise ValueError("incorrect value")
+Traceback (most recent call last):
+ValueError: invalid value

python-doctest/calculations/README.md

@@ -0,0 +1,24 @@
+# Functions to Perform Arithmetic Calculations
+
+The `calculations.py` Python module provides basic arithmetic operations, including sum, subtraction, multiplication, and division.
+
+Here are a few examples of how to use the functions in `calculations.py`:
+
+```python
+>>> import calculations
+
+>>> calculations.add(2, 2)
+4.0
+
+>>> calculations.subtract(2, 2)
+0.0
+
+>>> calculations.multiply(2, 2)
+4.0
+
+>>> calculations.divide(2, 2)
+1.0
+
+```
+
+These examples show how to use the `calculations.py` module in your code.

python-doctest/calculations/calculations.py

@@ -0,0 +1,65 @@
+"""Provide several sample math calculations.
+
+This module allows the user to make mathematical calculations.
+
+Module-level tests:
+>>> add(2, 4)
+6.0
+>>> subtract(5, 3)
+2.0
+>>> multiply(2.0, 4.0)
+8.0
+>>> divide(4.0, 2)
+2.0
+"""
+
+
+def add(a, b):
+    """Compute and return the sum of two numbers.
+
+    Tests for add():
+    >>> add(4.0, 2.0)
+    6.0
+    >>> add(4, 2)
+    6.0
+    """
+    return float(a + b)
+
+
+def subtract(a, b):
+    """Calculate the difference of two numbers.
+
+    Tests for subtract():
+    >>> subtract(4.0, 2.0)
+    2.0
+    >>> subtract(4, 2)
+    2.0
+    """
+    return float(a - b)
+
+
+def multiply(a, b):
+    """Compute and return the product of two numbers.
+
+    Tests for multiply():
+    >>> multiply(4.0, 2.0)
+    8.0
+    >>> multiply(4, 2)
+    8.0
+    """
+    return float(a * b)
+
+
+def divide(a, b):
+    """Compute and return the quotient of two numbers.
+
+    Tests for divide():
+    >>> divide(4.0, 2.0)
+    2.0
+    >>> divide(4, 2)
+    2.0
+    >>> divide(4, 0)
+    Traceback (most recent call last):
+    ZeroDivisionError: division by zero
+    """
+    return float(a / b)

python-doctest/calculations/test_claculations.py

@@ -0,0 +1,14 @@
+import doctest
+import unittest
+
+import calculations
+
+
+def load_tests(loader, tests, ignore):
+    tests.addTests(doctest.DocFileSuite("tests_file.txt"))
+    tests.addTests(doctest.DocTestSuite(calculations))
+    return tests
+
+
+if __name__ == '__main__':
+    unittest.main()

python-doctest/calculations/tests_file.txt

@@ -0,0 +1,13 @@
+>>> import calculations
+
+>>> calculations.add(2, 2)
+4.0
+
+>>> calculations.subtract(2, 2)
+0.0
+
+>>> calculations.multiply(2, 2)
+4.0
+
+>>> calculations.divide(2, 2)
+1.0

python-doctest/fizz_buzz.py

@@ -0,0 +1,72 @@
+# Replace numbers that are divisible by 3 with "fizz"
+def fizz_buzz(numbers):
+    """Implement the Fizz buzz game.
+    
+    >>> fizz_buzz([3, 6, 9, 12])
+    ['fizz', 'fizz', 'fizz', 'fizz']
+    """
+
+# # Replace numbers that are divisible by 3 with "fizz"
+# def fizz_buzz(numbers):
+#     """Implement the Fizz buzz game.
+    
+#     >>> fizz_buzz([3, 6, 9, 12])
+#     ['fizz', 'fizz', 'fizz', 'fizz']
+#     """
+#     result = []
+#     for number in numbers:
+#         if number % 3 == 0:
+#             result.append("fizz")
+#         else:
+#             result.append(number)
+#     return result
+
+
+# Replace numbers that are divisible by 5 with "buzz"
+# def fizz_buzz(numbers):
+#     """Implement the Fizz buzz game.
+    
+#     >>> fizz_buzz([3, 6, 9, 12])
+#     ['fizz', 'fizz', 'fizz', 'fizz']
+
+#     >>> fizz_buzz([5, 10, 20, 25])
+#     ['buzz', 'buzz', 'buzz', 'buzz']
+#     """
+#     result = []
+#     for number in numbers:
+#         if number % 3 == 0:
+#             result.append("fizz")
+#         elif number % 5 == 0:
+#             result.append("buzz")
+#         else:
+#             result.append(number)
+#     return result
+
+
+# Replace numbers that are divisible by 3 and 5 with "fizz buzz"
+# def fizz_buzz(numbers):
+#     """Implement the Fizz buzz game.
+    
+#     >>> fizz_buzz([3, 6, 9, 12])
+#     ['fizz', 'fizz', 'fizz', 'fizz']
+
+#     >>> fizz_buzz([5, 10, 20, 25])
+#     ['buzz', 'buzz', 'buzz', 'buzz']
+    
+#     >>> fizz_buzz([15, 30, 45])
+#     ['fizz buzz', 'fizz buzz', 'fizz buzz']
+
+#     >>> fizz_buzz([3, 6, 5, 10, 15, 30])
+#     ['fizz', 'fizz', 'buzz', 'buzz', 'fizz buzz', 'fizz buzz']
+#     """
+#     result = []
+#     for number in numbers:
+#         if number % 15 == 0:
+#             result.append("fizz buzz")
+#         elif number % 3 == 0:
+#             result.append("fizz")
+#         elif number % 5 == 0:
+#             result.append("buzz")
+#         else:
+#             result.append(number)
+#     return result

python-doctest/greet.py

@@ -0,0 +1,36 @@
+# def greet(name="World"):
+#     """Print a greeting.
+
+#     Usage examples:
+#     >>> greet("Pythonista")
+#     Hello, Pythonista!
+#     <BLANKLINE>
+#     How have you been?
+#     """
+#     print(f"Hello, {name}!")
+#     print()
+#     print("How have you been?")
+
+
+# def greet(name="World"):
+#     r"""Print a greeting.
+
+#     Usage examples:
+#     >>> greet("Pythonista")
+#     /== Hello, Pythonista! ==\
+#     \== How have you been? ==/
+#     """
+#     print(f"/== Hello, {name}! ==\\")
+#     print("\\== How have you been? ==/")
+
+
+def greet(name="World"):
+    """Print a greeting.
+
+    Usage examples:
+    >>> greet("Pythonista")
+    /== Hello, Pythonista! ==\\
+    \\== How have you been? ==/
+    """
+    print(f"/== Hello, {name}! ==\\")
+    print("\\== How have you been? ==/")

python-doctest/identity.py

@@ -0,0 +1,7 @@
+def get_id(obj):
+    """Return the identity of an object.
+    
+    >>> get_id(1)  # doctest: +ELLIPSIS
+    ...
+    """
+    return id(obj)

python-doctest/options.txt

@@ -0,0 +1,10 @@
+# options.txt
+
+>>> 5 < 7
+True
+
+>>> print("Hello, Pythonista! Welcome to Real Python!")
+Hello, ... Python!
+
+>>> print("\tHello, World!")
+	Hello, World!