d3-sankey-diagram/test/diagram-positionGroups-test.js

Sankey diagram d3 plugin

import positionGroup from '../src/positionGroup.js'

import test from 'tape'

test('positionGroup()', t => {
  const nodes = new Map([
    {
      id: 'a1',
      x0: 0,
      x1: 1,
      y0: 30,
      y1: 105
    }, {
      id: 'b',
      x0: 300,
      x1: 301,
      y0: 75,
      y1: 225
    }, {
      id: 'a2',
      x0: 0,
      x1: 1,
      y0: 195,
      y1: 270
    }
  ].map(d => [d.id, d]))

  const group1 = {
    title: 'Group',
    nodes: ['a1', 'a2']
  }

  const group2 = {
    title: 'B',
    nodes: ['b']
  }

  const group3 = {
    title: 'All',
    nodes: ['a1', 'a2', 'b']
  }

  t.deepEqual(positionGroup(nodes, group1), {
    title: 'Group',
    nodes: ['a1', 'a2'],
    rect: {
      top: 30,
      left: 0,
      bottom: 195 + 75,
      right: 1
    }
  }, 'group1')

  t.deepEqual(positionGroup(nodes, group2), {
    title: 'B',
    nodes: ['b'],
    rect: {
      top: 75,
      left: 300,
      bottom: 75 + 150,
      right: 301
    }
  }, 'group2')

  t.deepEqual(positionGroup(nodes, group3), {
    title: 'All',
    nodes: ['a1', 'a2', 'b'],
    rect: {
      top: 30,
      left: 0,
      bottom: 195 + 75,
      right: 301
    }
  }, 'group3')

  t.end()
})