import _ from "lodash";

import calcCentroid from "../../src/util/centroid";

import quantile from "../../src/util/quantile";
import * as Universe from "../../src/util/stateManager/universe";
import * as World from "../../src/util/stateManager/world";
import * as REST from "./stateManager/sampleResponses";
import { ControlsHelpers as CH } from "../../src/util/stateManager";

describe("centroid", () => {
  let world;
  let categoricalSelection;
  beforeAll(() => {
    // Create world + universe
    let universe = Universe.createUniverseFromResponse(
      _.cloneDeep(REST.config),
      _.cloneDeep(REST.schema)
    );

    universe = {
      ...universe,
      ...Universe.addObsAnnotations(
        universe,
        Universe.matrixFBSToDataframe(REST.annotationsObs)
      ),
      ...Universe.addVarAnnotations(
        universe,
        Universe.matrixFBSToDataframe(REST.annotationsVar)
      ),
      ...Universe.addObsLayout(
        universe,
        Universe.matrixFBSToDataframe(REST.layoutObs)
      )
    };
    world = World.createWorldFromEntireUniverse(universe);

    // Create categorical selection from world
    categoricalSelection = CH.createCategoricalSelection(
      world,
      CH.selectableCategoryNames(world.schema, CH.maxCategoryItems(REST.config))
    );
  });

  test("field4 (categorical obsAnnotation)", () => {
    const centroidResult = calcCentroid(
      world.obsAnnotations,
      world.obsLayout,
      "field4",
      ["umap_0", "umap_1"],
      categoricalSelection,
      world.schema.annotations.obsByName
    );

    // Check to see that a centroid has been calculated for every categorical value
    const keysAsArray = Array.from(centroidResult.keys());
    expect(keysAsArray).toEqual(
      expect.arrayContaining([83, true, "foo", 2.222222])
    );

    // This expected result assumes that all cells belong in all categorical values inside of sample response
    const expectedResult = [
      quantile([0.5], world.obsLayout.col("umap_0").asArray())[0],
      quantile([0.5], world.obsLayout.col("umap_1").asArray())[0]
    ];

    centroidResult.forEach(coordinate => {
      expect(coordinate).toEqual(expectedResult);
    });
  });

  test("field3 (boolean obsAnnotation)", () => {
    const centroidResult = calcCentroid(
      world.obsAnnotations,
      world.obsLayout,
      "field3",
      ["umap_0", "umap_1"],
      categoricalSelection,
      world.schema.annotations.obsByName
    );

    // Check to see that a centroid has been calculated for every categorical value
    const keysAsArray = Array.from(centroidResult.keys());
    expect(keysAsArray).toEqual(expect.arrayContaining([false, true]));

    // This expected result assumes that all cells belong in all categorical values inside of sample response
    const expectedResult = [
      quantile([0.5], world.obsLayout.col("umap_0").asArray())[0],
      quantile([0.5], world.obsLayout.col("umap_1").asArray())[0]
    ];

    centroidResult.forEach(coordinate => {
      expect(coordinate).toEqual(expectedResult);
    });
  });
});