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)