33 lines
1015 B
Python
33 lines
1015 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import os
|
|
from unittest import TestCase
|
|
|
|
import nbformat
|
|
from nbconvert.preprocessors import ExecutePreprocessor
|
|
|
|
TEST_DOC = os.getenv('TEST_DOC') == 'true'
|
|
notebook_paths = [
|
|
'./mmrazor/models/mutators/channel_mutator/channel_mutator.ipynb',
|
|
'./mmrazor/models/mutables/mutable_channel/units/mutable_channel_unit.ipynb', # noqa
|
|
'./demo/config_pruning.ipynb'
|
|
]
|
|
|
|
|
|
class TestDocs(TestCase):
|
|
|
|
def setUp(self) -> None:
|
|
if not TEST_DOC:
|
|
self.skipTest('disabled')
|
|
|
|
def test_notebooks(self):
|
|
for path in notebook_paths:
|
|
with self.subTest(path=path):
|
|
with open(path) as file:
|
|
nb_in = nbformat.read(file, nbformat.NO_CONVERT)
|
|
ep = ExecutePreprocessor(
|
|
timeout=600, kernel_name='python3')
|
|
try:
|
|
_ = ep.preprocess(nb_in)
|
|
except Exception:
|
|
self.fail()
|