final ouput from updating our math research project with cursor.

Author: mikeckennedy

code/09-agentic-ai/math-research/math_research.ipynb

@@ -12,7 +12,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 10,
    "id": "6040bded7e982e0c",
    "metadata": {
     "ExecuteTime": {
@@ -26,11 +26,13 @@
    "outputs": [],
    "source": [
     "# Import our math functions.\n",
-    "import mathf"
+    "import mathf\n",
+    "from itertools import islice"
    ]
   },
   {
-   "cell_type": "raw",
+   "cell_type": "markdown",
+   "id": "db59501c",
    "metadata": {
     "vscode": {
      "languageId": "raw"
@@ -83,8 +85,6 @@
    },
    "outputs": [],
    "source": [
-    "from itertools import islice\n",
-    "\n",
     "prime = 17\n",
     "all_fibs = mathf.fibonacci()\n",
     "fib_multiples_of_prime = mathf.multiples_of(all_fibs, prime)\n",

code/09-agentic-ai/math-research/test_mathf.py

@@ -0,0 +1,93 @@
+"""Unit tests for the mathf library."""
+
+import pytest
+from mathf import fibonacci, multiples_of
+
+
+class TestFibonacci:
+    """Tests for the fibonacci generator function."""
+
+    def test_fibonacci_first_ten_numbers(self):
+        """Test that fibonacci generates the correct first 10 numbers."""
+        fib = fibonacci()
+        expected = [1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
+        actual = [next(fib) for _ in range(10)]
+        assert actual == expected
+
+    def test_fibonacci_is_generator(self):
+        """Test that fibonacci returns a generator."""
+        fib = fibonacci()
+        assert hasattr(fib, '__next__')
+        assert hasattr(fib, '__iter__')
+
+    def test_fibonacci_continues_correctly(self):
+        """Test that fibonacci continues correctly after first few numbers."""
+        fib = fibonacci()
+        # Skip first 5 numbers
+        for _ in range(5):
+            next(fib)
+        # Test the next few
+        assert next(fib) == 8
+        assert next(fib) == 13
+        assert next(fib) == 21
+
+
+class TestMultiplesOf:
+    """Tests for the multiples_of function."""
+
+    def test_multiples_of_basic_case(self):
+        """Test basic functionality with multiples of 2."""
+        collection = [1, 2, 3, 4, 5, 6, 7, 8]
+        result = list(multiples_of(collection, 2))
+        assert result == [2, 4, 6, 8]
+
+    def test_multiples_of_three(self):
+        """Test with multiples of 3."""
+        collection = [1, 2, 3, 6, 9, 10, 12, 15]
+        result = list(multiples_of(collection, 3))
+        assert result == [3, 6, 9, 12, 15]
+
+    def test_multiples_of_one(self):
+        """Test with multiples of 1 (all numbers)."""
+        collection = [1, 2, 3, 4, 5]
+        result = list(multiples_of(collection, 1))
+        assert result == [1, 2, 3, 4, 5]
+
+    def test_multiples_of_empty_collection(self):
+        """Test with empty collection."""
+        result = list(multiples_of([], 2))
+        assert result == []
+
+    def test_multiples_of_no_matches(self):
+        """Test when no numbers are multiples."""
+        collection = [1, 3, 5, 7]
+        result = list(multiples_of(collection, 2))
+        assert result == []
+
+    def test_multiples_of_zero_raises_error(self):
+        """Test that zero raises ValueError."""
+        with pytest.raises(ValueError, match='Number must be positive'):
+            list(multiples_of([1, 2, 3], 0))
+
+    def test_multiples_of_negative_raises_error(self):
+        """Test that negative numbers raise ValueError."""
+        with pytest.raises(ValueError, match='Number must be positive'):
+            list(multiples_of([1, 2, 3], -5))
+
+    def test_multiples_of_is_generator(self):
+        """Test that multiples_of returns a generator."""
+        result = multiples_of([1, 2, 3, 4], 2)
+        assert hasattr(result, '__next__')
+        assert hasattr(result, '__iter__')
+
+    def test_multiples_of_with_large_numbers(self):
+        """Test with larger numbers."""
+        collection = [100, 150, 200, 250, 300]
+        result = list(multiples_of(collection, 50))
+        assert result == [100, 150, 200, 250, 300]
+
+    def test_multiples_of_with_duplicates(self):
+        """Test that duplicates in collection are preserved."""
+        collection = [2, 4, 2, 6, 4]
+        result = list(multiples_of(collection, 2))
+        assert result == [2, 4, 2, 6, 4]

requirements.piptools

@@ -1,2 +1,3 @@
 httpx
 jupyterlab
+pytest

requirements.txt

@@ -63,6 +63,8 @@ idna==3.10
     #   httpx
     #   jsonschema
     #   requests
+iniconfig==2.1.0
+    # via pytest
 ipykernel==6.30.0
     # via jupyterlab
 ipython==9.4.0
@@ -157,6 +159,7 @@ packaging==25.0
     #   jupyterlab
     #   jupyterlab-server
     #   nbconvert
+    #   pytest
 pandocfilters==1.5.1
     # via nbconvert
 parso==0.8.4
@@ -165,6 +168,8 @@ pexpect==4.9.0
     # via ipython
 platformdirs==4.3.8
     # via jupyter-core
+pluggy==1.6.0
+    # via pytest
 prometheus-client==0.22.1
     # via jupyter-server
 prompt-toolkit==3.0.51
@@ -184,6 +189,9 @@ pygments==2.19.2
     #   ipython
     #   ipython-pygments-lexers
     #   nbconvert
+    #   pytest
+pytest==8.4.1
+    # via -r requirements.piptools
 python-dateutil==2.9.0.post0
     # via
     #   arrow