@dpkit/table/field/types/yearmonth.spec.ts

Data Package implementation in TypeScript.

import { DataFrame, DataType, Series } from "nodejs-polars"
import { describe, expect, it } from "vitest"
import { normalizeTable } from "../../table/index.ts"
import { denormalizeTable } from "../../table/index.ts"

describe("parseYearmonthField", () => {
  it.each([
    ["2000-01", [2000, 1]],
    ["0-0", [0, 0]],
  ])("%s -> %s", async (cell, value) => {
    const table = DataFrame([Series("name", [cell], DataType.String)]).lazy()

    const schema = {
      fields: [{ name: "name", type: "yearmonth" as const }],
    }

    const ldf = await normalizeTable(table, schema)
    const df = await ldf.collect()

    expect(df.toRecords()[0]?.name).toEqual(value)
  })
})

describe("stringifyYearmonthField", () => {
  it.each([
    [[2000, 1], "2000-01"],
    [[2023, 12], "2023-12"],
    [[0, 0], "0000-00"],
  ])("%s -> %s", async (value, expected) => {
    const table = DataFrame([
      Series("name", [value], DataType.List(DataType.Int16)),
    ]).lazy()

    const schema = {
      fields: [{ name: "name", type: "yearmonth" as const }],
    }

    const ldf = await denormalizeTable(table, schema)
    const df = await ldf.collect()

    expect(df.toRecords()[0]?.name).toEqual(expected)
  })
})