Fix missing import in JS test

e2b-dev · Aug 28, 2024 · 08c2420 · 08c2420
1 parent a474640
commit 08c2420
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/js/tests/data.test.ts b/js/tests/data.test.ts
@@ -3,7 +3,10 @@ import { expect } from 'vitest'
 import { sandboxTest } from './setup'
 
 sandboxTest('get data', async ({ sandbox }) => {
-  const execution = await sandbox.notebook.execCell('pd.DataFrame({"a": [1, 2, 3]})')
+  const execution = await sandbox.notebook.execCell(`
+import pandas as pd
+pd.DataFrame({"a": [1, 2, 3]})
+`)
 
   const result = execution.results[0]
   expect(result.data).toBeDefined()