16 lines
1.1 KiB
JavaScript
16 lines
1.1 KiB
JavaScript
import test from 'node:test'
|
|
import assert from 'node:assert/strict'
|
|
import { stableMemoryArgs } from '../scripts/comfy-memory-policy.mjs'
|
|
|
|
test('memory policy preserves user launch flags and never forces a different VRAM loader', () => {
|
|
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' }), args)
|
|
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)
|
|
const defaults = ['main.py', '--enable-manager']
|
|
assert.deepEqual(stableMemoryArgs(defaults), defaults)
|
|
assert.notEqual(stableMemoryArgs(defaults), defaults)
|
|
assert.deepEqual(stableMemoryArgs(['main.py', '--disable-dynamic-vram']), ['main.py', '--disable-dynamic-vram'])
|
|
})
|