Repository navigation
Expand file tree
/
Copy pathtest_OpenMLDemo.py
More file actions
59 lines (48 loc) · 2.02 KB
/
Copy pathtest_OpenMLDemo.py
File metadata and controls
59 lines (48 loc) · 2.02 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import os
import unittest
import shutil
import sys
import nbformat
from nbconvert.preprocessors import ExecutePreprocessor
from nbconvert.preprocessors.execute import CellExecutionError
class OpenMLDemoTest(unittest.TestCase):
def setUp(self):
python_version = sys.version_info[0]
self.kernel_name = 'python%d' % python_version
self.this_file_directory = os.path.dirname(__file__)
self.notebook_output_directory = os.path.join(
self.this_file_directory, '.out')
try:
shutil.rmtree(self.notebook_output_directory)
except:
pass
try:
os.makedirs(self.notebook_output_directory)
except:
pass
def _test_notebook(self, notebook_name):
notebook_name = 'OpenMLDemo.ipynb'
notebook_filename = os.path.abspath(os.path.join(
self.this_file_directory, '..', '..', 'examples', notebook_name))
notebook_filename_out = os.path.join(
self.notebook_output_directory, notebook_name)
with open(notebook_filename) as f:
nb = nbformat.read(f, as_version=4)
nb.metadata.get('kernelspec', {})['name'] = self.kernel_name
ep = ExecutePreprocessor(kernel_name=self.kernel_name)
try:
ep.preprocess(nb, {'metadata': {'path': self.this_file_directory}})
except CellExecutionError as e:
msg = 'Error executing the notebook "%s". ' % notebook_filename
msg += 'See notebook "%s" for the traceback.\n\n' % notebook_filename_out
msg += e.traceback
self.fail(msg)
finally:
with open(notebook_filename_out, mode='wt') as f:
nbformat.write(nb, f)
@unittest.skip('SKIP for now until tests work again.')
def test_OpenMLDemo(self):
self._test_notebook('OpenMLDemo.ipynb')
@unittest.skip('SKIP for now until tests work again.')
def test_PyOpenML(self):
self._test_notebook('PyOpenML.ipynb')