@jrc03c/js-math-tools/src/clamp.test.mjs

some math tools for JS

import { clamp } from "./clamp.mjs"
import { DataFrame, Series } from "./dataframe/index.mjs"
import { expect, test } from "@jrc03c/fake-jest"
import { forEach } from "./for-each.mjs"
import { isEqual } from "./is-equal.mjs"
import { normal } from "./normal.mjs"

test("tests that values can be clamped correctly", () => {
  const s = new Series({ hello: normal(100) })
  const d = new DataFrame({ foo: normal(100), bar: normal(100) })

  const rights = [
    [0, 0],
    [1, 1],
    [2.3, 1],
    [-2.3, 0],
    [Infinity, 1],
    [-Infinity, 0],
    [
      [2, 3, 4],
      [1, 1, 1],
    ],
    [
      [
        [2, -3, 4],
        [-5, 6, -7],
      ],
      [
        [1, 0, 1],
        [0, 1, 0],
      ],
    ],
    [234n, 1n],
    [-234n, 0n],
    [
      [2n, -3n, 4n, -5n],
      [1n, 0n, 1n, 0n],
    ],
    [s, s.copy().apply(v => (v > 1 ? 1 : v < 0 ? 0 : v))],
    [d, d.copy().apply(col => col.apply(v => (v > 1 ? 1 : v < 0 ? 0 : v)))],
  ]

  forEach(rights, pair => {
    expect(isEqual(clamp(pair[0], 0, 1), pair[1])).toBe(true)
  })

  const wrongs = [
    NaN,
    "foo",
    true,
    false,
    null,
    undefined,
    Symbol.for("Hello, world!"),
    x => x,
    function (x) {
      return x
    },
    { hello: "world" },
  ]

  forEach(wrongs, v => {
    expect(clamp(v, 0, 1)).toBeNaN()
  })

  expect(
    isEqual(
      clamp(new Series(["a", "b", "c"]), 0, 1),
      new Series([NaN, NaN, NaN]),
    ),
  ).toBe(true)

  expect(
    isEqual(
      clamp(
        new DataFrame([
          ["a", "d"],
          ["b", "e"],
          ["c", "f"],
        ]),
      ),
      new DataFrame([
        [NaN, NaN],
        [NaN, NaN],
        [NaN, NaN],
      ]),
    ),
  ).toBe(true)
})