3 files changed, 2 insertions, 25 deletions
diff --git a/spec/frontend/notebook/cells/output/latex_spec.js b/spec/frontend/notebook/cells/output/latex_spec.js
index 848d2069421..ed3b63be50a 100644
--- a/spec/frontend/notebook/cells/output/latex_spec.js
+++ b/spec/frontend/notebook/cells/output/latex_spec.js
@@ -27,7 +27,7 @@ describe('LaTeX output cell', () => {
     ${1}  | ${false}
   `('sets `Prompt.show-output` to $expectation when index is $index', ({ index, expectation }) => {
     const wrapper = createComponent(inlineLatex, index);
-    const prompt = wrapper.find(Prompt);
+    const prompt = wrapper.findComponent(Prompt);
 
     expect(prompt.props().count).toEqual(count);
     expect(prompt.props().showOutput).toEqual(expectation);
diff --git a/spec/frontend/notebook/index_spec.js b/spec/frontend/notebook/index_spec.js
index 475c41a72f6..b79000a3505 100644
--- a/spec/frontend/notebook/index_spec.js
+++ b/spec/frontend/notebook/index_spec.js
@@ -11,7 +11,7 @@ describe('Notebook component', () => {
 
   function buildComponent(notebook) {
     return mount(Component, {
-      propsData: { notebook, codeCssClass: 'js-code-class' },
+      propsData: { notebook },
       provide: { relativeRawPath: '' },
     }).vm;
   }
@@ -46,10 +46,6 @@ describe('Notebook component', () => {
     it('renders code cell', () => {
       expect(vm.$el.querySelector('pre')).not.toBeNull();
     });
-
-    it('add code class to code blocks', () => {
-      expect(vm.$el.querySelector('.js-code-class')).not.toBeNull();
-    });
   });
 
   describe('with worksheets', () => {
@@ -72,9 +68,5 @@ describe('Notebook component', () => {
     it('renders code cell', () => {
       expect(vm.$el.querySelector('pre')).not.toBeNull();
     });
-
-    it('add code class to code blocks', () => {
-      expect(vm.$el.querySelector('.js-code-class')).not.toBeNull();
-    });
   });
 });
diff --git a/spec/frontend/notebook/lib/highlight_spec.js b/spec/frontend/notebook/lib/highlight_spec.js
deleted file mode 100644
index 944ccd6aa9f..00000000000
--- a/spec/frontend/notebook/lib/highlight_spec.js
+++ /dev/null
@@ -1,15 +0,0 @@
-import Prism from '~/notebook/lib/highlight';
-
-describe('Highlight library', () => {
-  it('imports python language', () => {
-    expect(Prism.languages.python).toBeDefined();
-  });
-
-  it('uses custom CSS classes', () => {
-    const el = document.createElement('div');
-    el.innerHTML = Prism.highlight('console.log("a");', Prism.languages.javascript);
-
-    expect(el.querySelector('.string')).not.toBeNull();
-    expect(el.querySelector('.function')).not.toBeNull();
-  });
-});