コード例 #1
0
ファイル: test_jupyter.py プロジェクト: vishalbelsare/kedro
 def test_renaming_default_kernel(self, mocker):
     """
     Make sure the default kernel display_name is changed.
     """
     mocker.patch.object(
         KernelSpecManager,
         "get_kernel_spec",
         return_value=mocker.Mock(display_name="default"),
     )
     manager = SingleKernelSpecManager()
     manager.default_kernel_name = "New Kernel Name"
     new_kernel_spec = manager.get_kernel_spec(NATIVE_KERNEL_NAME)
     assert new_kernel_spec.display_name == "New Kernel Name"
コード例 #2
0
ファイル: test_jupyter.py プロジェクト: vishalbelsare/kedro
 def test_non_default_kernel_untouched(self, mocker):
     """
     Make sure the non-default kernel display_name is not changed.
     In theory the function will never be called like that,
     but let's not make extra assumptions.
     """
     mocker.patch.object(
         KernelSpecManager,
         "get_kernel_spec",
         return_value=mocker.Mock(display_name="default"),
     )
     manager = SingleKernelSpecManager()
     manager.default_kernel_name = "New Kernel Name"
     new_kernel_spec = manager.get_kernel_spec("another_kernel")
     assert new_kernel_spec.display_name == "default"