mirror of
https://github.com/chanzuckerberg/cellxgene.git
synced 2026-09-29 16:48:11 +08:00
934cc5c69b
* Added TS. Updated build and linting config. Added types. * [ts-migrate][.] Rename files from JS/JSX to TS/TSX Co-authored-by: ts-migrate <> * [ts-migrate][.] Run TS Migrate Co-authored-by: ts-migrate <> * Corrected files mangled by ts-migrate. * Updated lint config, minor linting. * Re-enabled Husky. * Updated tests and config. * Reverted webpack devtool config. * Removed obsolete snapshots. * Added annotations snap. * Updated tsconfig includes wrt linting. * Removed ts-migrate. Co-authored-by: Timmy Huang <tihuan@users.noreply.github.com>
76 lines
2.5 KiB
TypeScript
76 lines
2.5 KiB
TypeScript
import cloneDeep from "lodash.clonedeep";
|
|
|
|
import calcCentroid from "../../src/util/centroid";
|
|
import quantile from "../../src/util/quantile";
|
|
import { matrixFBSToDataframe } from "../../src/util/stateManager/matrix";
|
|
import * as REST from "./stateManager/sampleResponses";
|
|
import { indexEntireSchema } from "../../src/util/stateManager/schemaHelpers";
|
|
import { _normalizeCategoricalSchema } from "../../src/annoMatrix/schema";
|
|
|
|
describe("centroid", () => {
|
|
let schema: any;
|
|
let obsAnnotations: any;
|
|
let obsLayout: any;
|
|
|
|
beforeAll(() => {
|
|
schema = indexEntireSchema(cloneDeep(REST.schema.schema));
|
|
obsAnnotations = matrixFBSToDataframe(REST.annotationsObs);
|
|
obsLayout = matrixFBSToDataframe(REST.layoutObs);
|
|
|
|
_normalizeCategoricalSchema(
|
|
schema.annotations.obsByName.field3,
|
|
obsAnnotations.col("field3")
|
|
);
|
|
});
|
|
|
|
test("field4 (categorical obsAnnotation)", () => {
|
|
const centroidResult = calcCentroid(
|
|
schema,
|
|
"field4",
|
|
obsAnnotations,
|
|
{ current: "umap", currentDimNames: ["umap_0", "umap_1"] },
|
|
obsLayout
|
|
);
|
|
|
|
// 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], obsLayout.col("umap_0").asArray())[0],
|
|
quantile([0.5], obsLayout.col("umap_1").asArray())[0],
|
|
];
|
|
|
|
centroidResult.forEach((coordinate: any) => {
|
|
expect(coordinate).toEqual(expectedResult);
|
|
});
|
|
});
|
|
|
|
test("field3 (boolean obsAnnotation)", () => {
|
|
const centroidResult = calcCentroid(
|
|
schema,
|
|
"field3",
|
|
obsAnnotations,
|
|
{ current: "umap", currentDimNames: ["umap_0", "umap_1"] },
|
|
obsLayout
|
|
);
|
|
|
|
// 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], obsLayout.col("umap_0").asArray())[0],
|
|
quantile([0.5], obsLayout.col("umap_1").asArray())[0],
|
|
];
|
|
|
|
centroidResult.forEach((coordinate: any) => {
|
|
expect(coordinate).toEqual(expectedResult);
|
|
});
|
|
});
|
|
});
|