Skip to content

Commit

Permalink
test: unit test for k-means
Browse files Browse the repository at this point in the history
  • Loading branch information
@zqqcee
zqqcee committed Oct 7, 2023
1 parent b6a8f3c commit 30d7c3b
Show file tree
Hide file tree
Showing 3 changed files with 419 additions and 1 deletion.
389 changes: 389 additions & 0 deletions __tests__/unit/k-means.spec.ts
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,389 @@
import { kMeans } from '../../packages/graph/src'
import propertiesGraphData from '../data/cluster-origin-properties-data.json';
import { Graph } from "@antv/graphlib";
import { dataPropertiesTransformer, dataLabelDataTransformer } from '../utils/data';


describe('kMeans abnormal demo', () => {
it('no properties demo: ', () => {
const noPropertiesData = {
nodes: [
{
id: 'node-0',
data: {},
},
{
id: 'node-1',
data: {},
},
{
id: 'node-2',
data: {},
},
{
id: 'node-3',
data: {},
}
],
}
const graph = new Graph(noPropertiesData);
const { clusters, clusterEdges } = kMeans(graph, 2);
expect(clusters.length).toBe(1);
expect(clusterEdges.length).toBe(0);
});
});


describe('kMeans normal demo', () => {
it('simple data demo: ', () => {
const simpleGraphData = {
nodes: [
{
id: 'node-0',
properties: {
amount: 10,
city: '10001',
}
},
{
id: 'node-1',
properties: {
amount: 10000,
city: '10002',
}
},
{
id: 'node-2',
properties: {
amount: 3000,
city: '10003',
}
},
{
id: 'node-3',
properties: {
amount: 3200,
city: '10003',
}
},
{
id: 'node-4',
properties: {
amount: 2000,
city: '10003',
}
}
],
edges: [
{
id: 'edge-0',
source: 'node-0',
target: 'node-1',
},
{
id: 'edge-1',
source: 'node-0',
target: 'node-2',
},
{
id: 'edge-4',
source: 'node-3',
target: 'node-2',
},
{
id: 'edge-5',
source: 'node-2',
target: 'node-1',
},
{
id: 'edge-6',
source: 'node-4',
target: 'node-1',
},
]
}
const data = dataPropertiesTransformer(simpleGraphData);
const graph = new Graph(data);
const { clusters } = kMeans(graph, 3);
expect(clusters.length).toBe(3);
const nodes = graph.getAllNodes();
expect(nodes[2].data.clusterId).toEqual(nodes[3].data.clusterId);
expect(nodes[2].data.clusterId).toEqual(nodes[4].data.clusterId);
});


it('complex data demo: ', () => {
const data = dataLabelDataTransformer(propertiesGraphData);
const graph = new Graph(data);
const { clusters } = kMeans(graph, 3);
expect(clusters.length).toBe(3);
const nodes = graph.getAllNodes();
expect(nodes[0].data.clusterId).toEqual(nodes[1].data.clusterId);
expect(nodes[0].data.clusterId).toEqual(nodes[2].data.clusterId);
expect(nodes[0].data.clusterId).toEqual(nodes[3].data.clusterId);
expect(nodes[0].data.clusterId).toEqual(nodes[4].data.clusterId);
expect(nodes[5].data.clusterId).toEqual(nodes[6].data.clusterId);
expect(nodes[5].data.clusterId).toEqual(nodes[7].data.clusterId);
expect(nodes[5].data.clusterId).toEqual(nodes[8].data.clusterId);
expect(nodes[5].data.clusterId).toEqual(nodes[9].data.clusterId);
expect(nodes[5].data.clusterId).toEqual(nodes[10].data.clusterId);
expect(nodes[11].data.clusterId).toEqual(nodes[12].data.clusterId);
expect(nodes[11].data.clusterId).toEqual(nodes[13].data.clusterId);
expect(nodes[11].data.clusterId).toEqual(nodes[14].data.clusterId);
expect(nodes[11].data.clusterId).toEqual(nodes[15].data.clusterId);
expect(nodes[11].data.clusterId).toEqual(nodes[16].data.clusterId);
});

it('demo use involvedKeys: ', () => {
const simpleGraphData = {
nodes: [
{
id: 'node-0',
properties: {
amount: 10,
city: '10001',
}
},
{
id: 'node-1',
properties: {
amount: 10000,
city: '10002',
}
},
{
id: 'node-2',
properties: {
amount: 3000,
city: '10003',
}
},
{
id: 'node-3',
properties: {
amount: 3200,
city: '10003',
}
},
{
id: 'node-4',
properties: {
amount: 2000,
city: '10003',
}
}
],
edges: [
{
id: 'edge-0',
source: 'node-0',
target: 'node-1',
},
{
id: 'edge-1',
source: 'node-0',
target: 'node-2',
},
{
id: 'edge-4',
source: 'node-3',
target: 'node-2',
},
{
id: 'edge-5',
source: 'node-2',
target: 'node-1',
},
{
id: 'edge-6',
source: 'node-4',
target: 'node-1',
},
]
}
const data = dataPropertiesTransformer(simpleGraphData);
const involvedKeys = ['amount'];
const graph = new Graph(data);
const { clusters } = kMeans(graph, 3, involvedKeys);
expect(clusters.length).toBe(3);
const nodes = graph.getAllNodes();
expect(nodes[2].data.clusterId).toEqual(nodes[3].data.clusterId);
expect(nodes[2].data.clusterId).toEqual(nodes[4].data.clusterId);
});

it('demo use uninvolvedKeys: ', () => {
const simpleGraphData = {
nodes: [
{
id: 'node-0',
properties: {
amount: 10,
city: '10001',
}
},
{
id: 'node-1',
properties: {
amount: 10000,
city: '10002',
}
},
{
id: 'node-2',
properties: {
amount: 3000,
city: '10003',
}
},
{
id: 'node-3',
properties: {
amount: 3200,
city: '10003',
}
},
{
id: 'node-4',
properties: {
amount: 2000,
city: '10003',
}
}
],
edges: [
{
id: 'edge-0',
source: 'node-0',
target: 'node-1',
},
{
id: 'edge-1',
source: 'node-0',
target: 'node-2',
},
{
id: 'edge-4',
source: 'node-3',
target: 'node-2',
},
{
id: 'edge-5',
source: 'node-2',
target: 'node-1',
},
{
id: 'edge-6',
source: 'node-4',
target: 'node-1',
},
]
}
const data = dataPropertiesTransformer(simpleGraphData);
const graph = new Graph(data);
const uninvolvedKeys = ['id', 'city'];
const { clusters } = kMeans(graph, 3, [], uninvolvedKeys);
expect(clusters.length).toBe(3);
const nodes = graph.getAllNodes(); data
expect(nodes[2].data.clusterId).toEqual(nodes[3].data.clusterId);
expect(nodes[2].data.clusterId).toEqual(nodes[4].data.clusterId);
});

});

describe('kMeans All properties values are numeric demo', () => {
it('all properties values are numeric demo: ', () => {
const allPropertiesValuesNumericData = {
nodes: [
{
id: 'node-0',
properties: {
max: 1000000,
mean: 900000,
min: 800000,
}
},
{
id: 'node-1',
properties: {
max: 1600000,
mean: 1100000,
min: 600000,
}
},
{
id: 'node-2',
properties: {
max: 5000,
mean: 3500,
min: 2000,
}
},
{
id: 'node-3',
properties: {
max: 9000,
mean: 7500,
min: 6000,
}
}
],
edges: [],
}
const data = dataPropertiesTransformer(allPropertiesValuesNumericData);
const graph = new Graph(data);
const { clusters, clusterEdges } = kMeans(graph, 2);
expect(clusters.length).toBe(2);
expect(clusterEdges.length).toBe(0);
const nodes = graph.getAllNodes();
expect(nodes[0].data.clusterId).toEqual(nodes[1].data.clusterId);
expect(nodes[2].data.clusterId).toEqual(nodes[3].data.clusterId);
});
it('only one property and the value are numeric demo: ', () => {
const allPropertiesValuesNumericData = {
nodes: [
{
id: 'node-0',
properties: {
num: 10,
}
},
{
id: 'node-1',
properties: {
num: 12,
}
},
{
id: 'node-2',
properties: {
num: 56,
}
},
{
id: 'node-3',
properties: {
num: 300,
}
},
{
id: 'node-4',
properties: {
num: 350,
}
}
],
edges: [],
}
const data = dataPropertiesTransformer(allPropertiesValuesNumericData);
const graph = new Graph(data);
const { clusters, clusterEdges } = kMeans(graph, 2);
expect(clusters.length).toBe(2);
expect(clusterEdges.length).toBe(0);
const nodes = graph.getAllNodes();
expect(nodes[0].data.clusterId).toEqual(nodes[1].data.clusterId);
expect(nodes[0].data.clusterId).toEqual(nodes[2].data.clusterId);
expect(nodes[3].data.clusterId).toEqual(nodes[4].data.clusterId);
});

});

Loading

0 comments on commit 30d7c3b

Please sign in to comment.