polls.test.js

const fs = require("fs");
const Papa = require("papaparse");

const [header, ...polls] = Papa.parse(
  fs.readFileSync("Data/Polls.csv", { encoding: "utf8" })
).data.filter((poll) => poll.length > 1);
const newerPolls = polls.slice(0, 70);
const allParties = ["M", "L", "C", "KD", "S", "V", "MP", "SD", "FI"];

test("Header is correct", () => {
  expect(header.length).toBe(18);
  expect(header).toContain("Company");
  expect(header).toContain("PublDate");
  expect(header).toContain("n");
  expect(header).toContain("collectPeriodFrom");
  expect(header).toContain("collectPeriodTo");
  allParties.forEach((party) => expect(header).toContain(party));
});

const companyIndex = header.indexOf("Company");
const publDateIndex = header.indexOf("PublDate");
const collectFromIndex = header.indexOf("collectPeriodFrom");
const collectToIndex = header.indexOf("collectPeriodTo");

const pollValue = (poll, party) => {
  const index = header.indexOf(party);
  if (index === -1) {
    throw new Error("Failed to find party " + party);
  }

  if (poll[index] === "NA") {
    return 0;
  }

  return parseFloat(poll[index]);
};

test("Contains polls", () => {
  expect(polls.length).toBeGreaterThan(1600);
});

test("Newer poll trends make sense", () => {
  newerPolls.forEach((poll) => {
    const parties = {};
    allParties.forEach((party) => {
      parties[party] = pollValue(poll, party);
    });

    expect(
      parties["S"],
      `S in (${poll}) has less than 24% support`
    ).toBeGreaterThan(24);
    expect(
      parties["S"],
      `S in (${poll}) has more than 38.7% support`
    ).toBeLessThan(38.7);

    expect(
      parties["V"],
      `V in (${poll}) has less than 6% support`
    ).toBeGreaterThan(6);
    expect(
      parties["V"],
      `V in (${poll}) has more than 13.5% support`
    ).toBeLessThan(13.5);

    expect(
      parties["MP"],
      `MP in (${poll}) has less than 2.3% support`
    ).toBeGreaterThan(2.3);
    expect(
      parties["MP"],
      `MP in (${poll}) has more than 8.2% support`
    ).toBeLessThan(8.2);

    expect(
      parties["M"],
      `M in (${poll}) has less than 14.7% support`
    ).toBeGreaterThan(14.7);
    expect(
      parties["M"],
      `M in (${poll}) has more than 24.5% support`
    ).toBeLessThan(24.6);

    expect(
      parties["L"],
      `L in (${poll}) has less than 1.8% support`
    ).toBeGreaterThan(1.7);
    expect(
      parties["L"],
      `L in (${poll}) has more than 4.7% support`
    ).toBeLessThan(4.7);

    expect(
      parties["KD"],
      `KD in (${poll}) has less than 2.3% support`
    ).toBeGreaterThan(2.3);
    expect(
      parties["KD"],
      `KD in (${poll}) has more than 6.2% support`
    ).toBeLessThan(6.2);

    expect(
      parties["C"],
      `C in (${poll}) has less than 3.2% support`
    ).toBeGreaterThan(3.2);
    expect(
      parties["C"],
      `C in (${poll}) has more than 8.9% support`
    ).toBeLessThan(8.9);

    expect(
      parties["SD"],
      `SD in (${poll}) has less than 15.5% support`
    ).toBeGreaterThan(15.4);
    expect(
      parties["SD"],
      `SD in (${poll}) has more than 30% support`
    ).toBeLessThan(30);
  });
});

test("All polls sum between 90% and 100%", () => {
  polls.forEach((poll) => {
    const values = allParties.map((party) => pollValue(poll, party));
    const sum = values.reduce((a, b) => a + b, 0);
    expect(
      sum,
      `Poll (${poll}) sums to ${sum}, more than 100%`
    ).toBeLessThanOrEqual(100);
    expect(sum, `Poll (${poll}) sums to ${sum}, less than 90%`).toBeGreaterThan(
      90
    );
  });
});

// test('Polls should be ordered by date', () => {
//   polls.forEach((poll) => {
//     const publicationDate = new Date(poll[pollIndex])
//     expect(sum, `Poll (${poll}) sums to ${sum}, more than 100%`).toBeLessThanOrEqual(100)
//     expect(sum, `Poll (${poll}) sums to ${sum}, less than 90%`).toBeGreaterThan(90)
//   })
// })

// test('Collection periods should not be too long', () => {
//   polls.forEach((poll) => {
//     const publicationDate = new Date(poll[pollIndex])
//     expect(sum, `Poll (${poll}) sums to ${sum}, more than 100%`).toBeLessThanOrEqual(100)
//     expect(sum, `Poll (${poll}) sums to ${sum}, less than 90%`).toBeGreaterThan(90)
//   })
// })