1import os
2
3from ..._compat import encodebytes
4from ..nbbase import (
5    NotebookNode,
6    new_code_cell, new_text_cell, new_worksheet, new_notebook, new_output,
7    new_metadata, new_author
8)
9
10# some random base64-encoded *bytes*
11png = encodebytes(os.urandom(5))
12jpeg = encodebytes(os.urandom(6))
13
14ws = new_worksheet(name='worksheet1')
15
16ws.cells.append(new_text_cell(
17    u'html',
18    source='Some NumPy Examples',
19    rendered='Some NumPy Examples'
20))
21
22
23ws.cells.append(new_code_cell(
24    input='import numpy',
25    prompt_number=1,
26    collapsed=False
27))
28
29ws.cells.append(new_text_cell(
30    u'markdown',
31    source='A random array',
32    rendered='A random array'
33))
34
35ws.cells.append(new_code_cell(
36    input='a = numpy.random.rand(100)',
37    prompt_number=2,
38    collapsed=True
39))
40
41ws.cells.append(new_code_cell(
42    input='print a',
43    prompt_number=3,
44    collapsed=False,
45    outputs=[new_output(
46        output_type=u'pyout',
47        output_text=u'<array a>',
48        output_html=u'The HTML rep',
49        output_latex=u'$a$',
50        output_png=png,
51        output_jpeg=jpeg,
52        output_svg=u'<svg>',
53        output_json=u'json data',
54        output_javascript=u'var i=0;',
55        prompt_number=3
56    ),new_output(
57        output_type=u'display_data',
58        output_text=u'<array a>',
59        output_html=u'The HTML rep',
60        output_latex=u'$a$',
61        output_png=png,
62        output_jpeg=jpeg,
63        output_svg=u'<svg>',
64        output_json=u'json data',
65        output_javascript=u'var i=0;'
66    ),new_output(
67        output_type=u'pyerr',
68        etype=u'NameError',
69        evalue=u'NameError was here',
70        traceback=[u'frame 0', u'frame 1', u'frame 2']
71    )]
72))
73
74authors = [new_author(name='Bart Simpson',email='bsimpson@fox.com',
75           affiliation=u'Fox',url=u'http://www.fox.com')]
76md = new_metadata(name=u'My Notebook',license=u'BSD',created=u'8601_goes_here',
77    modified=u'8601_goes_here',gistid=u'21341231',authors=authors)
78
79nb0 = new_notebook(
80    worksheets=[ws, new_worksheet(name='worksheet2')],
81    metadata=md
82)
83
84nb0_py = """# -*- coding: utf-8 -*-
85# <nbformat>2</nbformat>
86
87# <htmlcell>
88
89# Some NumPy Examples
90
91# <codecell>
92
93import numpy
94
95# <markdowncell>
96
97# A random array
98
99# <codecell>
100
101a = numpy.random.rand(100)
102
103# <codecell>
104
105print a
106
107"""
108
109
110