Make node def output class (#231)

* Refactor node def output

* Adjust test
This commit is contained in:
Chenlei Hu
2024-07-26 11:09:43 -04:00
committed by GitHub
parent 4dba1d3ab0
commit a1750212e5
3 changed files with 24 additions and 21 deletions

View File

@@ -171,7 +171,7 @@ describe('ComfyNodeDefImpl', () => {
expect(result.python_module).toBe('test_module')
expect(result.description).toBe('A test node')
expect(result.input).toBeInstanceOf(ComfyInputsSpec)
expect(result.output).toEqual({
expect(result.output.outputByName).toEqual({
intOutput: {
name: 'intOutput',
display_name: 'intOutput',
@@ -196,7 +196,7 @@ describe('ComfyNodeDefImpl', () => {
const result = plainToClass(ComfyNodeDefImpl, plainObject)
expect(result.output).toEqual({
expect(result.output.outputByName).toEqual({
stringOutput: {
name: 'stringOutput',
display_name: 'stringOutput',
@@ -234,7 +234,7 @@ describe('ComfyNodeDefImpl', () => {
const result = plainToClass(ComfyNodeDefImpl, plainObject)
expect(result.output).toEqual({
expect(result.output.outputByName).toEqual({
'0': {
name: '0',
display_name: 'INT',
@@ -289,7 +289,7 @@ describe('ComfyNodeDefImpl', () => {
const result = plainToClass(ComfyNodeDefImpl, plainObject)
expect(result.output).toEqual({})
expect(result.output.outputByName).toEqual({})
})
it('should handle complex input specifications', () => {