get_predicate implementation for vec

src/engine/array.rs

@@ -1,4 +1,4 @@
-use sprs::CsVec;
+use sprs::{CsVec, CsMat, CsMatBase};
 
 use crate::storage::ZarrArray;
 
@@ -10,8 +10,25 @@ impl EngineStrategy<CsVec<u8>> for ZarrArray {
         Ok(&self.transpose_view() * &selection)
     }
 
-    fn get_predicate(&self, index: usize) -> EngineResult<CsVec<u8>> {
-        unimplemented!()
+    fn get_predicate(&self, index: usize) -> EngineResult<CsMat<u8>> {
+
+
+        let mut rows: Vec<usize> = vec![];
+        let mut cols: Vec<usize> = vec![];
+        let mut values: Vec<u8> = vec![];
+        let iterator = self.into_iter();
+        for value in iterator {
+            if *value.0 == index as u8 {
+                rows.push(value.1.0);
+                cols.push(value.1.1);
+                values.push(*value.0);
+            }
+
+        }
+        //CsMatBase<u8, usize, Vec<usize>, Vec<usize>, Vec<u8>>
+        let result = CsMat::new(self.shape(),rows, cols, values);
+
+        Ok(result)
     }
 
     fn get_object(&self, index: usize) -> EngineResult<CsVec<u8>> {

tests/common/mod.rs

@@ -68,6 +68,7 @@ impl Predicate {
     pub fn get_idx(self, dictionary: &Dictionary) -> u8 {
         dictionary.get_predicate_idx_unchecked(self.into()) as u8
     }
+
 }
 
 impl From<Predicate> for &str {

tests/get_predicate_test.rs

@@ -0,0 +1,24 @@
+use remote_hdt::{
+    engine::EngineStrategy,
+    storage::{matrix::MatrixLayout, tabular::TabularLayout, ChunkingStrategy, LocalStorage},
+};
+use sprs::CsVec;
+
+mod common;
+
+#[test]
+fn get_predicate_tabular_test() {
+    let mut storage = LocalStorage::new(TabularLayout);
+    common::setup(common::TABULAR_ZARR, &mut storage, ChunkingStrategy::Chunk);
+
+    let actual = storage
+        .load_sparse(common::TABULAR_ZARR)
+        .unwrap()
+        .get_predicate(common::Predicate::InstanceOf.get_idx(&storage.get_dictionary()) as usize)
+        .unwrap();
+
+    assert_eq!(
+        actual,
+        CsVec::new(9, vec![0, 1, 2, 7, 8], vec![2, 4, 5, 7, 8])
+    )
+}