def test_split_code_and_text_blocks():
"""Test if a known example file gets properly split"""
blocks = sg.split_code_and_text_blocks('examples/just_code.py')
assert_equal(blocks[0][0], 'text')
assert_equal(blocks[1][0], 'code')
def test_bug_cases_of_notebook_syntax():
"""Test over the known requirements of supported syntax in the
notebook styled comments"""
with open('sphinxgallery/tests/reference_parse.txt') as reference:
ref_blocks = ast.literal_eval(reference.read())
blocks = sg.split_code_and_text_blocks('tutorials/plot_parse.py')
assert_equal(blocks, ref_blocks)
def test_direct_comment_after_docstring():
# For more details see
# https://github.com/sphinx-gallery/sphinx-gallery/pull/49
with tempfile.NamedTemporaryFile('w') as f:
f.write('\n'.join(['"Docstring"',
'# and now comes the module code',
'# with a second line of comment',
'x, y = 1, 2',
'']))
f.flush()
result = sg.split_code_and_text_blocks(f.name)
expected_result = [
('text', 'Docstring'),
('code', '\n'.join(['# and now comes the module code',
'# with a second line of comment',
'x, y = 1, 2',
'']))]
assert_equal(result, expected_result)
