fix: fix tree test

antvis · Sep 25, 2024 · 3cde03a · 3cde03a
1 parent ccd2cc6
commit 3cde03a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/packages/g6/__tests__/unit/utils/tree.spec.ts b/packages/g6/__tests__/unit/utils/tree.spec.ts
@@ -11,9 +11,10 @@ describe('tree', () => {
       nodes: [
         {
           id: 'root',
+          depth: 0,
           children: ['child'],
         },
-        { id: 'child' },
+        { id: 'child', depth: 1 },
       ],
       edges: [{ source: 'root', target: 'child' }],
     });
@@ -30,10 +31,11 @@ describe('tree', () => {
         {
           id: 'root',
           children: ['child'],
+          depth: 0,
           style: { fill: 'red' },
           data: { value: 10 },
         },
-        { id: 'child', style: { fill: 'green' }, data: { value: 1 } },
+        { id: 'child', depth: 1, style: { fill: 'green' }, data: { value: 1 } },
       ],
       edges: [{ source: 'root', target: 'child' }],
     });
@@ -59,10 +61,11 @@ describe('tree', () => {
         {
           id: 'root',
           children: ['child'],
+          depth: 0,
           style: { fill: 'red' },
           data: { value: 10 },
         },
-        { id: 'child', style: { fill: 'green' }, data: { value: 1 } },
+        { id: 'child', depth: 1, style: { fill: 'green' }, data: { value: 1 } },
       ],
       edges: [{ source: 'root', target: 'child', data: { weight: 11 } }],
     });