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)