mirror of
https://github.com/chanzuckerberg/cellxgene.git
synced 2026-09-19 19:08:11 +08:00
4efabf5231
* refactor to remove non-standard terms * switch key/value -> label/coords * fix missing parenth
97 lines
3.0 KiB
JavaScript
97 lines
3.0 KiB
JavaScript
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);
|
|
});
|
|
});
|
|
});
|