0a5f6a8047
- Go backend (server/)
- Frontend (web/, server/static/)
- Database and deployment files
- Scripts and docs
Co-Authored-By: 狸花猫/Claude-Qwen3.6-Plus 🐾
102 lines
2.6 KiB
TypeScript
102 lines
2.6 KiB
TypeScript
import G6, { GraphData } from '@antv/g6';
|
|
import { data } from './data';
|
|
import Bundling from '../../src/bundling';
|
|
|
|
const div = document.createElement('div');
|
|
div.id = 'force-layout';
|
|
document.body.appendChild(div);
|
|
|
|
describe('edge bundling', () => {
|
|
const graph = new G6.Graph({
|
|
container: div,
|
|
width: 500,
|
|
height: 500,
|
|
layout: {
|
|
type: 'circular',
|
|
},
|
|
defaultNode: { size: 10 },
|
|
});
|
|
|
|
graph.data(data);
|
|
graph.render();
|
|
|
|
it('edge bundling on circular layout with default configs', () => {
|
|
const bundle = new Bundling();
|
|
bundle.initPlugin(graph);
|
|
|
|
const graphData = graph.save() as GraphData;
|
|
bundle.bundling(graphData);
|
|
|
|
expect(graphData.edges[0].type).toEqual('polyline');
|
|
expect(graphData.edges[0].controlPoints.length > 2).toEqual(true);
|
|
bundle.destroy();
|
|
});
|
|
|
|
it('bundling on circular with fixed bundleThreshold and iterations', () => {
|
|
const bundle = new Bundling({
|
|
iterations: 120,
|
|
bundleThreshold: 0.1,
|
|
});
|
|
bundle.initPlugin(graph);
|
|
|
|
const graphData = graph.save() as GraphData;
|
|
bundle.bundling(graphData);
|
|
|
|
expect(graphData.edges[0].type).toEqual('polyline');
|
|
expect(graphData.edges[0].controlPoints.length > 2).toEqual(true);
|
|
bundle.destroy();
|
|
});
|
|
|
|
it('bundling update', () => {
|
|
const data2: GraphData = {
|
|
nodes: [{ id: 'n0' }, { id: 'n1' }],
|
|
edges: [{ source: 'n0', target: 'n1' }],
|
|
};
|
|
graph.changeData(data2);
|
|
graph.on('afterlayout', () => {
|
|
const bundle = new Bundling();
|
|
bundle.initPlugin(graph);
|
|
bundle.bundling(data2);
|
|
|
|
data2.nodes = [
|
|
{ id: 'n0', x: 10, y: 100 },
|
|
{ id: 'n1', x: 100, y: 100 },
|
|
{ id: 'n2', x: 10, y: 10 },
|
|
];
|
|
|
|
data2.edges = [
|
|
{ source: 'n0', target: 'n1' },
|
|
{ source: 'n1', target: 'n2' },
|
|
{ source: 'n0', target: 'n2' },
|
|
];
|
|
|
|
bundle.updateBundling({
|
|
bundleThreshold: 0.1,
|
|
iterations: 120,
|
|
data: data2,
|
|
});
|
|
|
|
expect(data2.edges[0].type).toEqual('polyline');
|
|
expect(data2.edges[0].controlPoints.length > 2).toEqual(true);
|
|
bundle.destroy();
|
|
});
|
|
});
|
|
|
|
it('bundling no position info, throw error', () => {
|
|
const bundle = new Bundling();
|
|
bundle.initPlugin(graph);
|
|
|
|
const data2: GraphData = {
|
|
nodes: [{ id: 'n0' }, { id: 'n1' }],
|
|
edges: [{ source: 'n0', target: 'n1' }],
|
|
};
|
|
|
|
function fn() {
|
|
bundle.bundling(data2);
|
|
}
|
|
expect(fn).toThrowError('please layout the graph or assign x and y for nodes first');
|
|
bundle.destroy();
|
|
graph.destroy();
|
|
});
|
|
});
|