Files
aigen/tests/comfy-memory-policy.test.mjs
T

12 lines
875 B
JavaScript

import test from 'node:test'
import assert from 'node:assert/strict'
import { stableMemoryArgs } from '../scripts/comfy-memory-policy.mjs'
test('Windows uses the supported non-dynamic loader without changing workflow or precision flags', () => {
const args = ['main.py', '--enable-dynamic-vram', '--fast-disk', '--enable-manager', '--fp16-vae']
assert.deepEqual(stableMemoryArgs(args, { platform: 'win32', cliSource: '--disable-dynamic-vram', dynamicVram: 'false' }), ['main.py', '--enable-manager', '--fp16-vae', '--disable-dynamic-vram'])
assert.deepEqual(stableMemoryArgs(args, { platform: 'win32', cliSource: '' }), args)
assert.deepEqual(stableMemoryArgs(args, { platform: 'linux', cliSource: '--disable-dynamic-vram' }), args)
assert.deepEqual(stableMemoryArgs(args, { platform: 'win32', cliSource: '--disable-dynamic-vram', dynamicVram: 'true' }), args)
})