def test_data_loader(self):
data = np.ones((25, 10)) * 100
paired = np.ones((25, 4)) * np.arange(0, 4)
pair_names = ["gabou", "achille", "pedro", "oclivio"]
y = CellMeasurement(name="dev",
data=paired,
columns_attr_name="dev_names",
columns=pair_names)
dataset = GeneExpressionDataset()
dataset.populate_from_data(data, Ys=[y])
ad = dataset.to_anndata()
dataset_ad = AnnDatasetFromAnnData(
ad, cell_measurements_col_mappings={"dev": "dev_names"})
self.assertTrue((paired == dataset_ad.dev).all())
self.assertTrue((dataset.X == dataset_ad.X).all())
self.assertTrue((dataset.cell_types == dataset_ad.cell_types).all())
def test_populate_from_data_with_measurements(self):
data = np.ones((25, 10)) * 100
paired = np.ones((25, 4)) * np.arange(0, 4)
pair_names = ["gabou", "achille", "pedro", "oclivio"]
y = CellMeasurement(name="dev",
data=paired,
columns_attr_name="dev_names",
columns=pair_names)
dataset = GeneExpressionDataset()
dataset.populate_from_data(data, Ys=[y])
self.assertEqual(dataset.nb_genes, 10)
self.assertEqual(dataset.nb_cells, 25)
self.assertTrue(hasattr(dataset, "dev"))
self.assertTrue(hasattr(dataset, "dev_names"))
self.assertListEqual(dataset.dev_names.tolist(), pair_names)
self.assertListEqual(dataset.dev[0].tolist(), [0, 1, 2, 3])
ad = dataset.to_anndata()
self.assertTrue("dev" in ad.obsm)
