test_nb_poly.py

# -*- coding: utf-8 -*-
"""
@brief      test log(time=82s)
"""
import os
import unittest
from pyquickhelper.loghelper import fLOG
from pyquickhelper.pycode import (
    add_missing_development_version, skipif_appveyor)
from pyquickhelper.ipythonhelper import test_notebook_execution_coverage
import mlinsights


class TestNotebookPolynomialFeatures(unittest.TestCase):

    def setUp(self):
        add_missing_development_version(["jyquickhelper"], __file__, hide=True)

    @skipif_appveyor('too long')
    def test_notebook_poly(self):
        fLOG(
            __file__,
            self._testMethodName,
            OutputPrint=__name__ == "__main__")

        self.assertTrue(mlinsights is not None)
        folder = os.path.join(os.path.dirname(__file__),
                              "..", "..", "_doc", "notebooks", "sklearn")
        test_notebook_execution_coverage(
            __file__, "poly", folder, 'mlinsights', fLOG=fLOG)


if __name__ == "__main__":
    unittest.main()