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"
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"