Repository: TheAlgorithms/Python

Issue:
Improve our test coverage
### Feature description

Many of our existing algorithm files have little to no unit testing. This is problematic because this can easily let bugs slip through. We want some assurance that the code we currently have is correct and functional. We welcome all contributors to open PRs to help us add tests to our codebase.

### How to find low-coverage files

Go to the Actions tab in this repository and find the most recent **build** workflow run. Open the logs under "Run Tests" and scroll down until you find the section on code coverage:
```
---------- coverage: platform linux, python 3.12.0-final-0 -----------
Name                                                                          Stmts   Miss  Cover   Missing
-----------------------------------------------------------------------------------------------------------
quantum/q_fourier_transform.py                                                   30     30     0%   14-93
scripts/validate_solutions.py                                                    54     54     0%   2-94
strings/min_cost_string_conversion.py                                            78     75     4%   20-57, 61-75, 79-129
...
```
The "Cover" column tells you what percentage of the lines in that file are covered by tests. We want to increase this percentage for existing files. Find a file with low coverage percentage that you wish to write tests for, add doctests for each function, and open a PR with your changes. You do not need to have a perfect coverage percentage, but all functions should have doctests.

Some files will naturally be hard to write tests for. For example, the file may be poorly written because they lack any functions. Other files might be how-tos, meaning they simply demonstrate how to use an existing library's functions rather than implementing the algorithm themselves. Ignore these kinds of files, as they will need to be rewritten eventually. Furthermore, ignore files in the `web_programming` and `project_euler` directories. Web programming files are inherently hard to test and Project Euler files have their own validation workflow, so don't worry about their test coverage.

_**When you open your PR, put "Contributes to #9943" in the PR description.**_ Do not use the word "fixes", "resolves", or "closes". This issue is an ongoing one, and your PR will not single-handedly resolve this issue.

### How to add doctests

A doctest is a unit test that is contained within the documentation comment (docstring) for a function. Here is an example of what doctests look like within a docstring:
```py
def add(a: int, b: int) -> int:
    """
    Adds two non-negative numbers.
    >>> add(1, 1)
    2
    >>> add(2, 5)
    7
    >>> add(1, 0)
    1
    >>> add(-1, -1)
    Traceback (most recent last):
    ...
    ValueError: Numbers must be non-negative
    """
```
For every function in the file you choose, you should write doctests like the ones shown above in its docstring. If a function doesn't have a docstring, add one. Your doctests should be comprehensive but not excessive: you should write just enough tests to cover all basic cases as well as all edge cases (e.g., negative numbers, empty lists, etc).

Do not simply run a function on some example inputs and put its output as the expected output for a doctest. This assumes that the function is implemented correctly when it might not be. Verify independently that your doctests and their expected outputs are correct. **Your PR will not be merged if it has failing tests.** If you happen to discover a bug while writing doctests, please fix it.

_**Please read our [contributing guidelines](https://github.com/TheAlgorithms/Python/blob/master/CONTRIBUTING.md) before you contribute.**_


Discussion:

Resolving patch:
diff --git a/data_structures/binary_tree/segment_tree.py b/data_structures/binary_tree/segment_tree.py
--- a/data_structures/binary_tree/segment_tree.py
+++ b/data_structures/binary_tree/segment_tree.py
@@ -3,7 +3,8 @@
 
 class SegmentTree:
     def __init__(self, a):
-        self.N = len(a)
+        self.A = a
+        self.N = len(self.A)
         self.st = [0] * (
             4 * self.N
         )  # approximate the overall size of segment tree with array N
@@ -11,14 +12,32 @@ def __init__(self, a):
             self.build(1, 0, self.N - 1)
 
     def left(self, idx):
+        """
+        Returns the left child index for a given index in a binary tree.
+
+        >>> s = SegmentTree([1, 2, 3])
+        >>> s.left(1)
+        2
+        >>> s.left(2)
+        4
+        """
         return idx * 2
 
     def right(self, idx):
+        """
+        Returns the right child index for a given index in a binary tree.
+
+        >>> s = SegmentTree([1, 2, 3])
+        >>> s.right(1)
+        3
+        >>> s.right(2)
+        5
+        """
         return idx * 2 + 1
 
     def build(self, idx, l, r):  # noqa: E741
         if l == r:
-            self.st[idx] = A[l]
+            self.st[idx] = self.A[l]
         else:
             mid = (l + r) // 2
             self.build(self.left(idx), l, mid)
@@ -26,6 +45,15 @@ def build(self, idx, l, r):  # noqa: E741
             self.st[idx] = max(self.st[self.left(idx)], self.st[self.right(idx)])
 
     def update(self, a, b, val):
+        """
+        Update the values in the segment tree in the range [a,b] with the given value.
+
+        >>> s = SegmentTree([1, 2, 3, 4, 5])
+        >>> s.update(2, 4, 10)
+        True
+        >>> s.query(1, 5)
+        10
+        """
         return self.update_recursive(1, 0, self.N - 1, a - 1, b - 1, val)
 
     def update_recursive(self, idx, l, r, a, b, val):  # noqa: E741
@@ -44,6 +72,15 @@ def update_recursive(self, idx, l, r, a, b, val):  # noqa: E741
         return True
 
     def query(self, a, b):
+        """
+        Query the maximum value in the range [a,b].
+
+        >>> s = SegmentTree([1, 2, 3, 4, 5])
+        >>> s.query(1, 3)
+        3
+        >>> s.query(1, 5)
+        5
+        """
         return self.query_recursive(1, 0, self.N - 1, a - 1, b - 1)
 
     def query_recursive(self, idx, l, r, a, b):  # noqa: E741


Test patch: