fork/cellxgene
1
0
Fork 0
mirror of https://github.com/chanzuckerberg/cellxgene.git synced 2026-09-30 02:48:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add obs index to label file (#928)

Browse Source 
* remove debug logging

* load and save obs/row index in label file

* lint

* update tests
This commit is contained in:
Bruce Martin
2019-09-19 09:16:01 -07:00
committed by GitHub
parent 197ea9214a
commit 109c9e70ec
5 changed files with 23 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/test/test_scanpy_engine.py
+3 -2
View File
@@ -248,9 +248,10 @@ class WritableAnnotationTest(unittest.TestCase):
res = self.data.annotation_put_fbs("obs", fbs)
self.assertEqual(res, json.dumps({"status": "OK"}))
self.assertTrue(path.exists(self.label_file))
df = pd.read_csv(self.label_file)
df = pd.read_csv(self.label_file, index_col=0)
self.assertEqual(df.shape, (n_rows, 2))
self.assertEqual(set(df.columns), set(['cat_A', 'cat_B']))
self.assertTrue(self.data.original_obs_index.equals(df.index))
self.assertTrue(np.all(df['cat_A'] == ['label_A' for l in range(0, n_rows)]))
self.assertTrue(np.all(df['cat_B'] == ['label_B' for l in range(0, n_rows)]))
@@ -262,7 +263,7 @@ class WritableAnnotationTest(unittest.TestCase):
res = self.data.annotation_put_fbs("obs", fbs)
self.assertEqual(res, json.dumps({"status": "OK"}))
self.assertTrue(path.exists(self.label_file))
df = pd.read_csv(self.label_file)
df = pd.read_csv(self.label_file, index_col=0)
self.assertEqual(set(df.columns), set(['cat_A', 'cat_C']))
self.assertTrue(np.all(df['cat_A'] == ['label_A1' for l in range(0, n_rows)]))
self.assertTrue(np.all(df['cat_C'] == ['label_C' for l in range(0, n_rows)]))