def test_thumbnail_number(): # which plot to show as the thumbnail image for test_str in ['# sphinx_gallery_thumbnail_number= 2', '# sphinx_gallery_thumbnail_number=2', '#sphinx_gallery_thumbnail_number = 2', ' # sphinx_gallery_thumbnail_number=2']: with tempfile.NamedTemporaryFile('w') as f: f.write('\n'.join(['"Docstring"', test_str])) f.flush() _, content = sg.get_docstring_and_rest(f.name) thumbnail_number = sg.extract_thumbnail_number(content) assert_equal(thumbnail_number, 2)
def test_thumbnail_number(): # which plot to show as the thumbnail image for test_str in [ '# sphinx_gallery_thumbnail_number= 2', '# sphinx_gallery_thumbnail_number=2', '#sphinx_gallery_thumbnail_number = 2', ' # sphinx_gallery_thumbnail_number=2' ]: with tempfile.NamedTemporaryFile('w') as f: f.write('\n'.join(['"Docstring"', test_str])) f.flush() _, content = sg.get_docstring_and_rest(f.name) thumbnail_number = sg.extract_thumbnail_number(content) assert_equal(thumbnail_number, 2)
def test_thumbnail_number(): # which plot to show as the thumbnail image for test_str in [ '# sphinx_gallery_thumbnail_number= 2', '# sphinx_gallery_thumbnail_number=2', '#sphinx_gallery_thumbnail_number = 2', ' # sphinx_gallery_thumbnail_number=2' ]: with tempfile.NamedTemporaryFile('w', delete=False) as f: f.write('\n'.join(['"Docstring"', test_str])) try: _, content = sg.get_docstring_and_rest(f.name) finally: os.remove(f.name) thumbnail_number = sg.extract_thumbnail_number(content) assert thumbnail_number == 2