summaryrefslogtreecommitdiff
path: root/tests/test_ipython.py
blob: 36e2e4b3c0a52c95b08fd6050c51941147621ce8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
"""
Test IPython/Jupyter display hooks
"""

import pytest

import pikepdf


@pytest.fixture
def graph(resources):
    return pikepdf.open(resources / 'graph.pdf')


def test_display_page(graph):
    page0 = graph.pages[0]
    mimebundle = page0._repr_mimebundle_(include=None, exclude=None)
    assert 'application/pdf' in mimebundle


def test_display_image(graph):
    im0 = graph.pages[0].Resources.XObject['/Im0']
    pim = pikepdf.PdfImage(im0)
    result = pim._repr_png_()
    assert result[1:4] == b'PNG'