netron/test/desktop.spec.js

import * as fs from 'fs';
import * as path from 'path';
import * as playwright from '@playwright/test';
import * as url from 'url';

playwright.test.setTimeout(120_000);

playwright.test('desktop', async () => {

    const self = url.fileURLToPath(import.meta.url);
    const dir = path.dirname(self);
    const file = path.resolve(dir, '../third_party/test/onnx/candy.onnx');
    playwright.expect(fs.existsSync(file)).toBeTruthy();

    // Launch app
    const electron = await playwright._electron;
    const args = ['.', '--no-sandbox'];
    const app = await electron.launch({ args });
    const window = await app.firstWindow();

    playwright.expect(window).toBeDefined();
    await window.waitForLoadState('domcontentloaded');

    const consent = await window.locator('#message-button');
    if (await consent.isVisible({ timeout: 2000 }).catch(() => false)) {
        await consent.click();
    }

    // Open the model
    await app.evaluate(async (electron, location) => {
        const windows = electron.BrowserWindow.getAllWindows();
        if (windows.length > 0) {
            const [window] = windows;
            window.webContents.send('open', { path: location });
        }
    }, file);

    // Wait for the graph to render
    await window.waitForSelector('#canvas', { state: 'attached', timeout: 10000 });
    await window.waitForSelector('body.default', { timeout: 10000 });

    // Open find sidebar
    await app.evaluate(async (electron) => {
        const windows = electron.BrowserWindow.getAllWindows();
        if (windows.length > 0) {
            const [window] = windows;
            window.webContents.send('find', {});
        }
    });
    await window.waitForTimeout(500);
    const search = await window.waitForSelector('#search', { state: 'visible', timeout: 5000 });
    playwright.expect(search).toBeDefined();

    // Find and activate tensor
    await search.fill('convolution1_W');
    await window.waitForSelector('.sidebar-find-content li', { state: 'attached' });
    const item = await window.waitForSelector('.sidebar-find-content li:has-text("convolution1_W")');
    await item.dblclick();

    // Expand the 'value' field
    const valueEntry = await window.waitForSelector('#sidebar-content .sidebar-item:has(.sidebar-item-name input[value="value"])');
    const valueButton = await valueEntry.waitForSelector('.sidebar-item-value-button');
    await valueButton.click();

    // Check first number from tensor value
    const pre = await valueEntry.waitForSelector('pre');
    const text = (await pre.textContent()) || '';
    const match = text.match(/-?\d+(?:\.\d+)?(?:e[+-]?\d+)?/i);
    playwright.expect(match).not.toBeNull();
    const first = parseFloat(match[0]);
    playwright.expect(first).toBe(0.1353299617767334);

    await app.close();
});