Added tests

CHANGELOG.md

@@ -1,6 +1,7 @@
 ## 0.2.2 (unreleased)
 
 - Added `p_iter` and `q_iter`
+- Added `p_row` and `q_col`
 
 ## 0.2.1 (2021-11-15)
 

README.md

@@ -44,7 +44,7 @@ model.p_factors();
 model.q_factors();
 ```
 
-Get the latent factors of a specific row or column
+Get the latent factors of a specific row or column [unreleased]
 
 ```rust
 model.p_row(row_index);

src/model.rs

@@ -177,6 +177,14 @@ mod tests {
         for (i, factors) in model.q_iter().enumerate() {
             assert_eq!(factors, q_vec[i]);
         }
+
+        assert_eq!(model.p_row(0), Some(p_vec[0]));
+        assert_eq!(model.p_row(1), Some(p_vec[1]));
+        assert_eq!(model.p_row(2), None);
+
+        assert_eq!(model.q_col(0), Some(q_vec[0]));
+        assert_eq!(model.q_col(1), Some(q_vec[1]));
+        assert_eq!(model.q_col(2), None);
     }
 
     #[test]