WIP: Migrate to ipytestsuite package

Author: edoardob90

Dockerfile

@@ -32,6 +32,8 @@ RUN mamba env update -n base -f /tmp/environment.yml && \
     echo "Installing CUDA-enabled PyTorch" && \
     pip install --no-cache-dir --force-reinstall torch torchvision; \
     fi && \
+    # Install ipytestsuite from TestPyPI (temporary, will use PyPI once published)
+    pip install --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ "ipytestsuite[ai]" && \
     # Clean up all package caches to reduce image size
     mamba clean --all -f -y && \
     # Remove pip cache

binder/environment.yml

@@ -4,7 +4,7 @@ channels:
   - conda-forge
   - pytorch
 dependencies:
-  - python=3.10
+  - python=3.12
   - pip
   - pip:
       - numpy
@@ -22,10 +22,6 @@ dependencies:
       - scikit-learn
       - attrs
       - multiprocess
-      - openai
-      - tenacity
-      - markdown2
-      - python-dotenv
       - pillow
       - opencv-python
       - torch

binder/postBuild

@@ -3,3 +3,6 @@ set -ex
 
 mkdir -p ${HOME}/.ipython/profile_default
 cp binder/ipython_config.py ${HOME}/.ipython/profile_default/
+
+# Install ipytestsuite from TestPyPI (temporary, will use PyPI once published)
+pip install --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ "ipytestsuite[ai]"

docker/environment.yml

@@ -3,6 +3,7 @@ name: base
 channels:
   - conda-forge
 dependencies:
+  - python=3.12
   - pip
   - pip:
       - numpy
@@ -20,10 +21,6 @@ dependencies:
       - scikit-learn
       - attrs
       - multiprocess
-      - openai
-      - tenacity
-      - markdown2
-      - python-dotenv
       - pillow
       - opencv-python
       - albumentations

magic_example.ipynb

@@ -2,24 +2,64 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {
     "tags": []
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div style='background-color: #d9ead3; border-radius: 5px; padding: 10px;'>✅ <strong>OpenAI client configured successfully.</strong></div>"
+      ],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "<div style='background-color: #fffde7; border-radius: 5px; padding: 10px;'>🔄 <strong>IPytest extension (re)loaded</strong> (with AI support).</div>"
+      ],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
    "source": [
-    "%reload_ext tutorial.tests.testsuite"
+    "%reload_ext ipytestsuite"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 3,
    "metadata": {
     "tags": []
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "36143cb14d384b5598bab2994c67954e",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "VBox(children=(Output(outputs=({'output_type': 'display_data', 'data': {'text/plain': 'HTML(value=\\'<div style…"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
    "source": [
-    "%%ipytest\n",
+    "%%ipytest --path tutorial/tests\n",
     "# or %%ipytest test_module_name\n",
     "\n",
     "def solution_power2(x: int) -> int:\n",
@@ -89,7 +129,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.10"
+   "version": "3.12.11"
   },
   "vscode": {
    "interpreter": {

pyproject.toml

@@ -2,7 +2,7 @@
 name = "python-tutorial"
 version = "v2025.05"
 description = "Jupyter-based Python programming tutorial"
-requires-python = ">=3.10"
+requires-python = ">=3.12"
 
 [project.optional-dependencies]
 dev = [
@@ -21,16 +21,13 @@ dev = [
   "scikit-learn",
   "attrs",
   "multiprocess",
-  "openai",
-  "tenacity",
-  "markdown2",
-  "python-dotenv",
   "pillow",
   "opencv-python",
   "albumentations",
   "grad-cam",
   "plotly",
-  "torch==2.7.0",  # pinned to last stable version as of 2025-05-14
+  "torch>=2.0.0",  # PyTorch 2.x series
+  "ipytestsuite[ai]>=0.1.0",
 ]
 
 # pytest
@@ -93,3 +90,11 @@ indent-style = "space"
 # Setuptools: suppress package discovery
 [tool.setuptools]
 packages = []
+
+[[tool.uv.index]]
+name = "testpypi"
+url = "https://test.pypi.org/simple/"
+explicit = true
+
+[tool.uv.sources]
+ipytestsuite = { index = "testpypi" }