Fix error with shape of sliced objects (CellArr#75)

Author: jkanche

CHANGELOG.md

@@ -1,6 +1,6 @@
 # Changelog
 
-## Version 0.5.1
+## Version 0.5.1 - 0.5.2
 
 - Support csc matrices in layers, although not common ran into a situation where the
 anndata object was stored from R.

src/cellarr/CellArrDatasetSlice.py

@@ -83,7 +83,7 @@ def to_summarizedexperiment(self):
 
     @property
     def shape(self):
-        return self.matrix.shape
+        return (len(self.cell_metadata), len(self.gene_annotation))
 
     def __len__(self):
         return self.shape[0]

tests/test_query.py

@@ -120,3 +120,11 @@ def test_query_cellarrdataset():
     for cell, cell_chunk in cd.itercells():
         cell_count += 1
     assert cell_count == 1100
+
+    full_row_slice = cd[1,]
+    assert full_row_slice is not None
+    assert full_row_slice.shape == (1, 1000)
+
+    full_col_slice = cd[:, 1]
+    assert full_col_slice is not None
+    assert full_col_slice.shape == (1100, 1)