Show More
@@ -1,4 +1,4 b'' | |||||
1 | from nbconvert import ConverterRST, main |
|
1 | from nbconvert import ConverterRST, main, converters | |
2 | import nose.tools as nt |
|
2 | import nose.tools as nt | |
3 |
|
3 | |||
4 | import os |
|
4 | import os | |
@@ -24,20 +24,29 b' class TestSimple(object):' | |||||
24 | shutil.rmtree(self.wrkdir) |
|
24 | shutil.rmtree(self.wrkdir) | |
25 |
|
25 | |||
26 | def outfile_exists(self, fmt): |
|
26 | def outfile_exists(self, fmt): | |
|
27 | extension = converters[fmt].extension | |||
27 | return os.path.exists(os.path.join(self.wrkdir, |
|
28 | return os.path.exists(os.path.join(self.wrkdir, | |
28 |
self.basename + '.' + |
|
29 | self.basename + '.' + extension)) | |
29 |
|
30 | |||
30 | def test_simple(self): |
|
31 | def test_simple(self): | |
31 | c = ConverterRST(self.infile) |
|
32 | c = ConverterRST(self.infile) | |
32 | f = c.render() |
|
33 | f = c.render() | |
33 | nt.assert_true(f.endswith('.rst'), 'changed file extension to rst') |
|
34 | nt.assert_true(f.endswith('.rst'), 'changed file extension to rst') | |
34 |
|
35 | |||
|
36 | def run_main(self, fmt): | |||
|
37 | """ | |||
|
38 | Run the 'main' method to convert the input file to the given | |||
|
39 | format and check that the expected output file exists. | |||
|
40 | """ | |||
|
41 | main(self.infile, format=fmt) | |||
|
42 | nt.assert_true(self.outfile_exists(fmt)) | |||
|
43 | ||||
35 | def test_main(self): |
|
44 | def test_main(self): | |
36 | """ |
|
45 | """ | |
37 | Test main entry point |
|
46 | Test main entry point with all known formats. | |
38 | """ |
|
47 | """ | |
39 | main(self.infile) |
|
48 | for fmt in converters: | |
40 | nt.assert_true(self.outfile_exists('rst')) |
|
49 | yield self.run_main, fmt | |
41 |
|
50 | |||
42 | def test_render_heading(self): |
|
51 | def test_render_heading(self): | |
43 | """ |
|
52 | """ | |
@@ -73,10 +82,3 b' class TestSimple(object):' | |||||
73 | chk_str = "Test for heading type H{0}\n{1}\n".format( |
|
82 | chk_str = "Test for heading type H{0}\n{1}\n".format( | |
74 | level, c.heading_level[level] * 24) |
|
83 | level, c.heading_level[level] * 24) | |
75 | nt.assert_equal(rst_str, chk_str) |
|
84 | nt.assert_equal(rst_str, chk_str) | |
76 |
|
||||
77 | def test_main_html(self): |
|
|||
78 | """ |
|
|||
79 | Test main entry point |
|
|||
80 | """ |
|
|||
81 | main(self.infile, format='html') |
|
|||
82 | nt.assert_true(self.outfile_exists('html')) |
|
General Comments 0
You need to be logged in to leave comments.
Login now