Function testCount

packages/db-ivm/tests/operators/count.test.ts:18–214

()

})

function testCount() {
  test(`basic count operation`, () => {
    const graph = new D2()
    const input = graph.newInput<[number, string]>()
    const tracker = new KeyedMessageTracker<number, number>()

    input.pipe(
      count(),
      output((message) => {
        tracker.addMessage(message)
      }),
    )

    graph.finalize()

    input.sendData(
      new MultiSet([
        [[1, `a`], 2],
        [[2, `b`], 1],
        [[2, `c`], 1],
        [[2, `d`], 1],
        [[3, `x`], 1],
        [[3, `y`], -1],
      ]),
    )
    input.sendData(new MultiSet([[[3, `z`], 1]]))
    graph.run()

    const result = tracker.getResult()

    // Assert only keys that have values are affected
    assertOnlyKeysAffected(`basic count operation`, result.messages, [1, 2, 3])

    // Assert the final materialized results are correct
    assertKeyedResults(
      `basic count operation`,
      result,
      [
        [1, 2], // 2 values for key 1
        [2, 3], // 3 values for key 2
        [3, 1], // 1 value for key 3 (1 + (-1) + 1 = 1)
      ],
      6, // Expected message count
    )
  })

  test(`count with all negative multiplicities`, () => {
    const graph = new D2()
    const input = graph.newInput<[number, string]>()
    const tracker = new KeyedMessageTracker<number, number>()

    input.pipe(
      count(),
      output((message) => {
        tracker.addMessage(message)
      }),
    )