__tests__/count.ts - main - githubmirror/immutable-js

__tests__/count.ts
80 строк2 KB
Julien Deniau
Better TypeScript configuration for jest
29 мар 2025, 01:47
29 мар 2025, 01:47992f37c
import { describe, expect, it } from '@jest/globals';
import { Range, Seq } from 'immutable';

describe('count', () => {
  it('counts sequences with known lengths', () => {
    expect(Seq([1, 2, 3, 4, 5]).size).toBe(5);
    expect(Seq([1, 2, 3, 4, 5]).count()).toBe(5);
  });

  it('counts sequences with unknown lengths, resulting in a cached size', () => {
    const seq = Seq([1, 2, 3, 4, 5, 6]).filter((x) => x % 2 === 0);
    expect(seq.size).toBe(undefined);
    expect(seq.count()).toBe(3);
    expect(seq.size).toBe(3);
  });

  it('counts sequences with a specific predicate', () => {
    const seq = Seq([1, 2, 3, 4, 5, 6]);
    expect(seq.size).toBe(6);
    expect(seq.count((x) => x > 3)).toBe(3);
  });

  describe('countBy', () => {
    it('counts by keyed sequence', () => {
      const grouped = Seq({ a: 1, b: 2, c: 3, d: 4 }).countBy((x) => x % 2);
      expect(grouped.toJS()).toEqual({ 1: 2, 0: 2 });
      expect(grouped.get(1)).toEqual(2);
    });

    it('counts by indexed sequence', () => {
      expect(
        Seq([1, 2, 3, 4, 5, 6])
          .countBy((x) => x % 2)
          .toJS()
      ).toEqual({ 1: 3, 0: 3 });
    });

    it('counts by specific keys', () => {
      expect(
        Seq([1, 2, 3, 4, 5, 6])
          .countBy((x) => (x % 2 ? 'odd' : 'even'))
          .toJS()
      ).toEqual({ odd: 3, even: 3 });
    });
  });

  describe('isEmpty', () => {
    it('is O(1) on sequences with known lengths', () => {
      expect(Seq([1, 2, 3, 4, 5]).size).toBe(5);
      expect(Seq([1, 2, 3, 4, 5]).isEmpty()).toBe(false);
      expect(Seq().size).toBe(0);
      expect(Seq().isEmpty()).toBe(true);
    });

    it('lazily evaluates Seq with unknown length', () => {
      let seq = Seq([1, 2, 3, 4, 5, 6]).filter((x) => x % 2 === 0);
      expect(seq.size).toBe(undefined);
      expect(seq.isEmpty()).toBe(false);
      expect(seq.size).toBe(undefined);

      seq = Seq([1, 2, 3, 4, 5, 6]).filter((x) => x > 10);
      expect(seq.size).toBe(undefined);
      expect(seq.isEmpty()).toBe(true);
      expect(seq.size).toBe(undefined);
    });

    it('with infinitely long sequences of known length', () => {
      const seq = Range(0, Infinity);
      expect(seq.size).toBe(Infinity);
      expect(seq.isEmpty()).toBe(false);
    });

    it('with infinitely long sequences of unknown length', () => {
      const seq = Range(0, Infinity).filter((x) => x % 2 === 0);
      expect(seq.size).toBe(undefined);
      expect(seq.isEmpty()).toBe(false);
      expect(seq.size).toBe(undefined);
    });
  });
});