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