Deploy AI apps for free on Ploomber Cloud!

pytest integration

pytest integration#

You can use ploomber-engine to test your notebooks with pytest, here’s a full example:


from pathlib import Path

import pytest
from ploomber_engine.ipython import PloomberClient

def ns():
    client = PloomberClient.from_path(Path(__file__).parent / 'nb.ipynb')
    return client.get_namespace()

def test_add(ns):
    add = ns['add']
    assert add(1, 2) == 3

def test_multiply(ns):
    multiply = ns['multiply']
    assert multiply(2, 2) == 4

def test_notebook_results(ns):
    assert ns['a'] == 42
    assert ns['b'] == 200

To run your tests, execute: