| 123456789101112131415161718192021222324252627 |
- # -*- coding: utf-8 -*-
- from jupyter_client.kernelspec import KernelSpecManager
- class TestKernelSpecManager(KernelSpecManager):
- """ A custom KernelSpecManager for testing.
- """
- def find_kernel_specs(self):
- """ Returns a dict mapping kernel names to resource directories.
- """
- kspecs = super(TestKernelSpecManager, self).find_kernel_specs()
- # add conda envs kernelspecs
- kspecs.update({name: spec.resource_dir
- for name, spec
- in self._kspecs.items()})
- return kspecs
- def get_kernel_spec(self, kernel_name):
- """ Returns a :class:`KernelSpec` instance for the given kernel_name.
- """
- return (
- self._kspecs.get(kernel_name) or
- super(TestKernelSpecManager, self).get_kernel_spec(kernel_name)
- )
|
