d3-sankey-diagram/test/sortNodes/index-test.js

Sankey diagram d3 plugin

import sortNodes from '../../src/sortNodes/index.js'
import tape from 'tape'
import { Graph } from '@dagrejs/graphlib'

tape('sortNodes()', test => {
  //
  //  a -- b -- d
  //   `-- c -- e
  //
  const G = new Graph({ directed: true, multigraph: true })
  G.setNode('a', { rank: 0 })
  G.setNode('b', { rank: 1 })
  G.setNode('c', { rank: 1 })
  G.setNode('d', { rank: 2 })
  G.setNode('e', { rank: 2 })
  G.setEdge('a', 'b', {})
  G.setEdge('a', 'c', {})
  G.setEdge('b', 'd', {})
  G.setEdge('c', 'e', {})
  sortNodes(G)

  test.deepEqual(depths(G), {
    a: 0,
    b: 0,
    c: 1,
    d: 0,
    e: 1
  })

  test.end()
})

function depths (G) {
  const r = {}
  G.nodes().forEach(u => { r[u] = G.node(u).depth })
  return r
}