chore: add schema types (#2369)

This commit is contained in:
Timmy Huang
2021-08-04 13:16:18 -07:00
committed by GitHub
parent 95ce39f2e9
commit 26de334274
8 changed files with 157 additions and 82 deletions
@@ -5,6 +5,7 @@ import zip from "lodash.zip";
import _ from "lodash";
import { flatbuffers } from "flatbuffers";
import { NetEncoding } from "../../../src/util/stateManager/matrix_generated";
import { RawSchema } from "../../../src/common/types/entities";
/*
test data mocking REST 0.2 API responses. Used in several tests.
@@ -29,7 +30,7 @@ const aConfigResponse = {
},
};
const aSchemaResponse = {
const aSchemaResponse: { schema: RawSchema } = {
schema: {
dataframe: {
nObs,
@@ -40,28 +41,30 @@ const aSchemaResponse = {
obs: {
index: "name",
columns: [
{ name: "name", type: "string" },
{ name: "field1", type: "int32" },
{ name: "field2", type: "float32" },
{ name: "field3", type: "boolean" },
{ name: "name", type: "string", writable: false },
{ name: "field1", type: "int32", writable: false },
{ name: "field2", type: "float32", writable: false },
{ name: "field3", type: "boolean", writable: false },
{
name: "field4",
type: "categorical",
categories: field4Categories,
writable: false,
},
],
},
var: {
index: "name",
columns: [
{ name: "name", type: "string" },
{ name: "fieldA", type: "int32" },
{ name: "fieldB", type: "float32" },
{ name: "fieldC", type: "boolean" },
{ name: "name", type: "string", writable: false },
{ name: "fieldA", type: "int32", writable: false },
{ name: "fieldB", type: "float32", writable: false },
{ name: "fieldC", type: "boolean", writable: false },
{
name: "fieldD",
type: "categorical",
categories: fieldDCategories,
writable: false,
},
],
},