|
4 | 4 | import { expect, test } from '@playwright/test';
|
5 | 5 | import { addBenchmarkToTest } from './utils';
|
6 | 6 |
|
7 |
| -const PREFIX = |
8 |
| - process.env.PROGRESSIVE_RENDERING === 'true' ? 'progressive_rendering_' : ''; |
| 7 | +const PROGRESSIVE_RENDERING = process.env.PROGRESSIVE_RENDERING === 'true'; |
| 8 | +const PREFIX = PROGRESSIVE_RENDERING ? '' : ''; |
9 | 9 | test.describe('Voila performance Tests', () => {
|
10 | 10 | test.beforeEach(({ page }) => {
|
11 | 11 | page.setDefaultTimeout(120000);
|
@@ -310,19 +310,31 @@ test.describe('Voila performance Tests', () => {
|
310 | 310 | const notebookName = 'query-strings';
|
311 | 311 | const testFunction = async () => {
|
312 | 312 | await page.goto(`/voila/render/${notebookName}.ipynb`);
|
313 |
| - await page.waitForSelector( |
314 |
| - 'div.lm-Widget[data-mime-type="application/vnd.jupyter.stdout"]' |
315 |
| - ); |
| 313 | + if (PROGRESSIVE_RENDERING) { |
| 314 | + await page.waitForSelector( |
| 315 | + 'div.lm-Widget[data-mime-type="application/vnd.jupyter.stdout"]' |
| 316 | + ); |
| 317 | + } else { |
| 318 | + await page.waitForSelector( |
| 319 | + 'div.jp-OutputArea-output[data-mime-type="text/plain"]' |
| 320 | + ); |
| 321 | + } |
316 | 322 | const userName = await page.$$(
|
317 | 323 | 'div.jp-RenderedText.jp-OutputArea-output > pre'
|
318 | 324 | );
|
319 | 325 | expect(await userName[1].innerHTML()).toContain('Hi Kim');
|
320 | 326 | };
|
321 | 327 | await addBenchmarkToTest(notebookName, testFunction, testInfo, browserName);
|
322 | 328 | await page.goto(`/voila/render/${notebookName}.ipynb?username=Riley`);
|
323 |
| - await page.waitForSelector( |
324 |
| - 'div.lm-Widget[data-mime-type="application/vnd.jupyter.stdout"]' |
325 |
| - ); |
| 329 | + if (PROGRESSIVE_RENDERING) { |
| 330 | + await page.waitForSelector( |
| 331 | + 'div.lm-Widget[data-mime-type="application/vnd.jupyter.stdout"]' |
| 332 | + ); |
| 333 | + } else { |
| 334 | + await page.waitForSelector( |
| 335 | + 'div.jp-OutputArea-output[data-mime-type="text/plain"]' |
| 336 | + ); |
| 337 | + } |
326 | 338 | const userName = await page.$$(
|
327 | 339 | 'div.jp-RenderedText.jp-OutputArea-output > pre'
|
328 | 340 | );
|
|
0 commit comments