Function test_notebook_export_single_display

tests/test_magic.py:1117–1161

Test that multiple MIME types create a single display_data output, not multiple.

()

def test_notebook_export_single_display():
    """Test that multiple MIME types create a single display_data output, not multiple."""
    pytest.importorskip("nbformat")

    _ip = get_ipython()
    orig_outputs = _ip.history_manager.outputs.copy()
    orig_execution_count = _ip.execution_count
    _ip.history_manager.reset()

    try:
        execution_count = _ip.execution_count = 1
        _ip.run_cell("'test'", store_history=True, silent=False)

        # Mock display output with multiple MIME types
        test_display_history = HistoryOutput(
            output_type="display_data",
            bundle={"text/plain": "test", "text/html": "<div>test</div>"},
        )
        _ip.history_manager.outputs[execution_count] = [test_display_history]

        with TemporaryDirectory() as td:
            outfile = f"{td}/test.ipynb"
            _ip.run_cell(f"%notebook {outfile}", store_history=True, silent=False)

            # Verify single display_data output with both MIME types
            with open(outfile, "r") as f:
                nb = json.load(f)

        cell = nb["cells"][0]
        display_outputs = [
            out for out in cell["outputs"] if out["output_type"] == "display_data"
        ]

        assert (
            len(display_outputs) == 1
        ), f"Expected 1 display_data output, got {len(display_outputs)}"

        output_data = display_outputs[0]["data"]
        assert set(output_data.keys()) == {"text/plain", "text/html"}
        assert output_data["text/plain"] == ["test"]
        assert output_data["text/html"] == ["<div>test</div>"]

    finally:
        _ip.history_manager.outputs = orig_outputs
        _ip.execution_count = orig_execution_count


class TestEnv(TestCase):