chore(example): add switch

example/src/Stress/index.js

@@ -33,15 +33,23 @@ const StressFlow = () => {
     });
   };
 
+  const updateElements = () => {
+    const grid = Math.ceil(Math.random() * 10);
+    setElements(getElements(grid, grid));
+  };
+
   return (
     <ReactFlow elements={elements} onLoad={onLoad} onElementsRemove={onElementsRemove} onConnect={onConnect}>
       <MiniMap />
       <Controls />
       <Background />
 
-      <button onClick={updatePos} style={{ position: 'absolute', right: 10, top: 30, zIndex: 4 }}>
-        change pos
-      </button>
+      <div style={{ position: 'absolute', right: 10, top: 10, zIndex: 4 }}>
+        <button onClick={updatePos} style={{ marginRight: 5 }}>
+          change pos
+        </button>
+        <button onClick={updateElements}>update elements</button>
+      </div>
     </ReactFlow>
   );
 };

example/src/Stress/utils.js

@@ -15,7 +15,7 @@ export function getElements(xElements = 10, yElements = 10) {
       };
       initialElements.push(node);
 
-      if (recentNodeId && nodeId <= (xElements * yElements)) {
+      if (recentNodeId && nodeId <= xElements * yElements) {
         initialElements.push({ id: `${x}-${y}`, source: recentNodeId.toString(), target: nodeId.toString() });
       }