d3-sankey-diagram/test/sortNodes/sort-by-positions-test.js

Sankey diagram d3 plugin

import sortByPositions from '../../src/sortNodes/sort-by-positions.js'
import { map } from 'd3-collection'
import tape from 'tape'

tape('sortByPositions', test => {
  let arr

  arr = ['a', 'b', 'c']
  sortByPositions(arr, map({ a: 0, b: 2, c: 1 }))
  test.deepEqual(arr, ['a', 'c', 'b'],
    'sorts according to given order')

  arr = ['a', 'b', 'c']
  sortByPositions(arr, map({ a: 2, b: 1, c: 1 }))
  test.deepEqual(arr, ['b', 'c', 'a'],
    'stable sort')

  arr = ['a', 'b', 'c']
  sortByPositions(arr, map({ a: 1, b: -1, c: 0 }))
  test.deepEqual(arr, ['c', 'b', 'a'],
    '-1 means stay in same position')

  test.end()
})